From a1619bcd04c7caf6f5c85e985af8ea1254529d5b Mon Sep 17 00:00:00 2001 From: Dominik Roth Date: Mon, 10 Jul 2023 10:57:50 +0200 Subject: [PATCH] wandb fixes --- slate/slate.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) 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: