Update callbacks.py

Former-commit-id: 44ebe58083dc62128fd14df474c11c6e09af43db
This commit is contained in:
hiyouga 2023-07-15 17:39:16 +08:00
parent e0d9f6f2b7
commit 6d4a107546

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