dodox/Slate
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

wandb fixes

Browse Source
This commit is contained in:
Dominik Moritz Roth 2023-07-10 10:57:50 +02:00
parent 4efa081d73
commit a1619bcd04
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
slate/slate.py
View File

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