🐛 Style Transfer: redo config update to add state in the config dict

arkhn · Nov 7, 2024 · 9cb46ad · 9cb46ad
1 parent 9a057ae
commit 9cb46ad
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/lib/style-transfer/style_transfer/rb_gen/steps/sft.py b/lib/style-transfer/style_transfer/rb_gen/steps/sft.py
@@ -60,6 +60,7 @@ def sft_train(
 
     cfg.sft.training_args.output_dir = f"models/{wandb.run.id}/sft"
     args = hydra.utils.instantiate(cfg.sft.training_args)
+    wandb.config.update({"state": "sft"}, allow_val_change=True)
     test_sft_dataset = None
     if cfg.dataset.sft_dataset is not None:
         sft_dataset, test_sft_dataset = sft_dataset.train_test_split(