riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-08-03 03:31:05 +06:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Fix swanlab global step (#36728)

Browse Source 
* fix

* global step
This commit is contained in:
Ze-Yi LIN 2025-03-20 18:13:37 +08:00 committed by GitHub
parent b815fae359
commit 8733297b41
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/transformers/integrations/integration_utils.py
View File

@ -2291,10 +2291,10 @@ class SwanLabCallback(TrainerCallback):
if state.is_world_process_zero:
for k, v in logs.items():
if k in single_value_scalars:
self._swanlab.log({f"single_value/{k}": v})
self._swanlab.log({f"single_value/{k}": v}, step=state.global_step)
non_scalar_logs = {k: v for k, v in logs.items() if k not in single_value_scalars}
non_scalar_logs = rewrite_logs(non_scalar_logs)
self._swanlab.log({**non_scalar_logs, "train/global_step": state.global_step})
self._swanlab.log({**non_scalar_logs, "train/global_step": state.global_step}, step=state.global_step)
def on_save(self, args, state, control, **kwargs):
if self._log_model is not None and self._initialized and state.is_world_process_zero: