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

bwuni is annoying

Browse Source
This commit is contained in:
Dominik Moritz Roth 2023-07-09 17:47:06 +02:00
parent 4e5f10378a
commit 733286e810
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
slate/slate.py
View File

@ -13,6 +13,7 @@ import pdb
d = pdb.set_trace d = pdb.set_trace
REQUIRE_CONFIG_CONSUMED = False REQUIRE_CONFIG_CONSUMED = False
WANDB_START_METHOD = 'process'
Parallelization_Primitive = Thread # Process Parallelization_Primitive = Thread # Process
@ -216,7 +217,7 @@ class Slate():
sweep_id = wandb.sweep( sweep_id = wandb.sweep(
sweep=sweepC, sweep=sweepC,
project=project, project=project,
settings=wandb.Settings(start_method="thread") settings=wandb.Settings(start_method=WANDB_START_METHOD)
) )
wandb.agent(sweep_id, function=partial(self._run_from_sweep, config, p_ind=p_ind), count=len(rep_ids)) wandb.agent(sweep_id, function=partial(self._run_from_sweep, config, p_ind=p_ind), count=len(rep_ids))
else: else:
@ -235,7 +236,7 @@ class Slate():
with wandb.init( with wandb.init(
project=project, project=project,
config=copy.deepcopy(config), config=copy.deepcopy(config),
settings=wandb.Settings(start_method="thread"), settings=wandb.Settings(start_method=WANDB_START_METHOD),
**wandbC **wandbC
) as run: ) as run:
runner(self, run, config) runner(self, run, config)
@ -256,7 +257,7 @@ class Slate():
with wandb.init( with wandb.init(
project=project, project=project,
settings=wandb.Settings(start_method="thread"), settings=wandb.Settings(start_method=WANDB_START_METHOD),
**wandbC **wandbC
) as run: ) as run:
config = copy.deepcopy(orig_config) config = copy.deepcopy(orig_config)