[tracker] Add Trackio Integration for LlamaFactory (#10165)

Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>
This commit is contained in:
Parag Ekbote
2026-03-03 14:49:37 +05:30
committed by GitHub
parent b5cb7cb0e6
commit eb976d75a2
3 changed files with 85 additions and 3 deletions

View File

@@ -371,6 +371,18 @@ class ReporterCallback(TrainerCallback):
}
)
if "trackio" in args.report_to:
import trackio
trackio.config.update(
{
"model_args": self.model_args.to_dict(),
"data_args": self.data_args.to_dict(),
"finetuning_args": self.finetuning_args.to_dict(),
"generating_args": self.generating_args.to_dict(),
}
)
if self.finetuning_args.use_swanlab:
import swanlab # type: ignore