wandb fixes

slate/slate.py

@@ -14,6 +14,7 @@ d = pdb.set_trace
 
 REQUIRE_CONFIG_CONSUMED = False
 WANDB_START_METHOD = 'fork'
+REINIT = True
 
 Parallelization_Primitive = Thread  # Process
 
@@ -236,7 +237,7 @@ class Slate():
             with wandb.init(
                 project=project,
                 config=copy.deepcopy(config),
-                reinit=True,
+                reinit=REINIT,
                 settings=wandb.Settings(start_method=WANDB_START_METHOD),
                 **wandbC
             ) as run:
@@ -258,7 +259,7 @@ class Slate():
 
         with wandb.init(
             project=project,
-            reinit=True,
+            reinit=REINIT,
             settings=wandb.Settings(start_method=WANDB_START_METHOD),
             **wandbC
         ) as run: