From 51b7abe9ddaee6619e18f111fc80afeb4d217304 Mon Sep 17 00:00:00 2001 From: YeAnbang Date: Thu, 12 Jun 2025 15:06:01 +0800 Subject: [PATCH] fix num_update_per_episode --- applications/ColossalChat/coati/distributed/launch.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/applications/ColossalChat/coati/distributed/launch.py b/applications/ColossalChat/coati/distributed/launch.py index 30e7382ef..dc8bf0057 100644 --- a/applications/ColossalChat/coati/distributed/launch.py +++ b/applications/ColossalChat/coati/distributed/launch.py @@ -16,7 +16,7 @@ def get_jsonl_size_fast(path: str) -> int: with open(path) as f: lines = f.readlines() lines = [line for line in lines if line.strip()] - return len(lines) - 1 + return len(lines) def get_dp_size_fast(n_procs: int, plugin_config: Dict[str, Any]) -> int: