Update callbacks.py

Former-commit-id: d640c5545f6b2fcf7840df2b35a80f49a2de4ba5
This commit is contained in:
hiyouga 2023-07-15 17:39:16 +08:00
parent 7c0f2ab0a4
commit 8ba0996a53

View File

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