fix callback

This commit is contained in:
hiyouga committed 2023-07-15 17:18:16 +08:00
1 parent f751376613
commit 22d9a9c2af
4 files changed
+9 -5

No files matched your search

+1 -1
View File
@@ -46,7 +46,7 @@ class LogCallback(TrainerCallback):
r"""
Event called after logging the last logs.
"""
if "current_steps" not in state.log_history[-1]:
if "step" not in state.log_history[-1]:
return
cur_time = time.time()
cur_steps = state.log_history[-1].get("step")