From 98ce7300863ea72baa54a5ef66f1fa1b202e2365 Mon Sep 17 00:00:00 2001 From: Dominik Roth Date: Thu, 27 Jul 2023 14:39:05 +0200 Subject: [PATCH] Trying to work around wandb quirks --- slate/slate.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/slate/slate.py b/slate/slate.py index 2622a88..b52af3d 100644 --- a/slate/slate.py +++ b/slate/slate.py @@ -19,7 +19,7 @@ REQUIRE_CONFIG_CONSUMED = False DEFAULT_START_METHOD = 'fork' DEFAULT_REINIT = True -Parallelization_Primitive = Thread # Process +Parallelization_Primitive = Process # Thread try: import pyslurm @@ -282,7 +282,7 @@ class Slate(): project=project, config=copy.deepcopy(config), reinit=self.consume(wandbC, 'reinit', DEFAULT_REINIT), - settings=wandb.Settings(start_method=self.consume(wandbC, 'start_method', DEFAULT_START_METHOD)), + settings=wandb.Settings(self.consume(wandbC, 'settings')), **wandbC ) as run: runner = Runner(self, config) @@ -306,7 +306,7 @@ class Slate(): with wandb.init( project=project, reinit=self.consume(wandbC, 'reinit', DEFAULT_REINIT), - settings=wandb.Settings(start_method=self.consume(wandbC, 'start_method', DEFAULT_START_METHOD)), + settings=wandb.Settings(self.consume(wandbC, 'settings')), **wandbC ) as run: config = copy.deepcopy(orig_config)