Skip to content

Commit 0e69b98

Browse files
authored
Merge pull request #6347 from hpcaitech/hotfix/fix_num_update_per_episode
[fix] fix num update per episode
2 parents ac06935 + 51b7abe commit 0e69b98

File tree

1 file changed

+1
-1
lines changed
  • applications/ColossalChat/coati/distributed

1 file changed

+1
-1
lines changed

applications/ColossalChat/coati/distributed/launch.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@ def get_jsonl_size_fast(path: str) -> int:
1616
with open(path) as f:
1717
lines = f.readlines()
1818
lines = [line for line in lines if line.strip()]
19-
return len(lines) - 1
19+
return len(lines)
2020

2121

2222
def get_dp_size_fast(n_procs: int, plugin_config: Dict[str, Any]) -> int:

0 commit comments

Comments
 (0)