feat: support wandb logging

2026-04-08 22:35:09 +00:00 · 2023-04-20 01:41:12 +09:00
parent 334589af4e
commit 27ffd9fe3d
6 changed files with 33 additions and 7 deletions
--- a/fine_tune.py
+++ b/fine_tune.py
@@ -260,7 +260,7 @@ def train(args):
    )

    if accelerator.is_main_process:
-        accelerator.init_trackers("finetuning")
+        accelerator.init_trackers("finetuning" if args.log_tracker_name is None else args.log_tracker_name)

    for epoch in range(num_train_epochs):
        print(f"epoch {epoch+1}/{num_train_epochs}")