Update callbacks.py

Former-commit-id: 44ebe58083dc62128fd14df474c11c6e09af43db
This commit is contained in:
hiyouga 2023-07-15 17:39:16 +08:00
parent 51368f3453
commit e9fe48150c

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