fix #176
This commit is contained in:
1 parent
6f9360c0bd
commit
cadeac0f44
3 files changed
+35
-18
No files matched your search
@@ -47,6 +47,9 @@ class LogCallback(TrainerCallback):
|
||||
r"""
|
||||
Event called after logging the last logs.
|
||||
"""
|
||||
if not state.is_world_process_zero:
|
||||
return
|
||||
|
||||
cur_time = time.time()
|
||||
cur_steps = state.log_history[-1].get("step")
|
||||
elapsed_time = cur_time - self.start_time
|
||||
|
||||
Reference in new issue
Block a user