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)