mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-07-24 03:59:33 +00:00
fix num_update_per_episode
This commit is contained in:
parent
ac069357a9
commit
51b7abe9dd
@ -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:
|
||||
|
Loading…
Reference in New Issue
Block a user