423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-12-17 04:10:36 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix callback

Browse Source
This commit is contained in:
hiyouga
2023-07-15 17:18:16 +08:00
parent f751376613
commit 22d9a9c2af
4 changed files with 9 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/llmtuner/extras/callbacks.py
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")