wandb tweaks

2023-07-09 18:06:06 +02:00 · 2023-07-09 18:06:06 +02:00 · 4efa081d73
commit 4efa081d73
parent a9bde894b8
1 changed files with 2 additions and 0 deletions
--- a/slate/slate.py
+++ b/slate/slate.py
@ -236,6 +236,7 @@ class Slate():
            with wandb.init(
                project=project,
                config=copy.deepcopy(config),
+                reinit=True,
                settings=wandb.Settings(start_method=WANDB_START_METHOD),
                **wandbC
            ) as run:
@ -257,6 +258,7 @@ class Slate():

        with wandb.init(
            project=project,
+            reinit=True,
            settings=wandb.Settings(start_method=WANDB_START_METHOD),
            **wandbC
        ) as run: