diff --git a/slate/slate.py b/slate/slate.py index 46db35c..6d11221 100644 --- a/slate/slate.py +++ b/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: