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

CHatGPT tells me i need to init BEFORE I sweep

Browse Source
This commit is contained in:
Dominik Moritz Roth 2023-07-27 16:38:24 +02:00
parent f484e9d24c
commit 373346b589
1 changed files with 24 additions and 23 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
slate/slate.py
View File

@ -263,8 +263,16 @@ class Slate():
def _run_process(self, orig_config, rep_ids, p_ind):
config = copy.deepcopy(orig_config)
if self.consume(config, 'sweep.enable', False):
wandbC = copy.deepcopy(config['wandb'])
wandb.agent(self.sweep_id, function=partial(self._run_from_sweep, config, p_ind=p_ind), count=len(rep_ids))
wandbC = self.consume(copy.deepcopy(orig_config), 'wandb', {}, expand=True)
project = self.consume(wandbC, 'project')
with wandb.init(
project=project,
reinit=self.consume(wandbC, 'reinit', DEFAULT_REINIT),
settings=wandb.Settings(**self.consume(wandbC, 'settings', {})),
**wandbC
) as run:
wandb.agent(self.sweep_id, function=partial(self._run_from_sweep, config, run, p_ind=p_ind), count=len(rep_ids))
else:
self.consume(config, 'sweep', {})
self._run_single(config, rep_ids, p_ind=p_ind)
@ -297,18 +305,11 @@ class Slate():
print(msg)
orig_config = config
def _run_from_sweep(self, orig_config, p_ind):
runnerName, wandbC = self.consume(orig_config, 'runner'), self.consume(orig_config, 'wandb', {}, expand=True)
project = self.consume(wandbC, 'project')
def _run_from_sweep(self, orig_config, run, p_ind):
runnerName = self.consume(orig_config, 'runner')
Runner = self.runners[runnerName]
with wandb.init(
project=project,
reinit=self.consume(wandbC, 'reinit', DEFAULT_REINIT),
settings=wandb.Settings(**self.consume(wandbC, 'settings', {})),
**wandbC
) as run:
config = copy.deepcopy(orig_config)
self.deep_update(config, wandb.config)
run.config = copy.deepcopy(config)