add entity (for v4)

segment-any-text · May 12, 2024 · 57545b1 · 57545b1
1 parent 66b97f4
commit 57545b1
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/wtpsplit/train/train_adapter.py b/wtpsplit/train/train_adapter.py
@@ -374,7 +374,8 @@ def maybe_pad(text):
 
     # 1 wandb run for all language-dataset combinations
     if "wandb" in training_args.report_to and training_args.process_index == 0:
-        wandb.init(name=wandb_name, project=args.wandb_project, group=wandb_name)
+        # TODO: don't hardcode entity
+        wandb.init(name=wandb_name, project=args.wandb_project, group=wandb_name, entity="markus_583")
         wandb.config.update(args)
         wandb.config.update(training_args)
         wandb.config.update(label_args)