fix bug in wandb init for finetune

mozilla-ai · Jan 19, 2024 · 3f0c8f0 · 3f0c8f0
1 parent cd68801
commit 3f0c8f0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/flamingo/jobs/finetuning/entrypoint.py b/src/flamingo/jobs/finetuning/entrypoint.py
@@ -122,7 +122,7 @@ def load_and_train(config: FinetuningJobConfig):
 def train_func(config_data: dict):
     config = FinetuningJobConfig(**config_data)
     if is_tracking_enabled(config):
-        with wandb_init_from_config(config, resume="never"):
+        with wandb_init_from_config(config.tracking, resume="never"):
             load_and_train(config)
     else:
         load_and_train(config)