fix num_update_per_episode

2025-09-07 12:01:39 +00:00 · 2025-06-12 15:06:01 +08:00
parent ac069357a9
commit 51b7abe9dd
1 changed files with 1 additions and 1 deletions
--- 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: