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

maybe fix wandb sweeps

Browse Source
This commit is contained in:
Dominik Moritz Roth 2023-07-12 18:02:10 +02:00
parent df0ca53a0a
commit 47c0b7b2c8
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
slate/slate.py
View File

@ -183,8 +183,8 @@ class Slate():
schedC = self.consume(config, 'scheduler') schedC = self.consume(config, 'scheduler')
s_name = self.consume(slurmC, 'name') s_name = self.consume(slurmC, 'name')
# Pre Validation
runnerName, _ = self.consume(config, 'runner'), self.consume(config, 'wandb', {}, expand=True) runnerName, _ = self.consume(config, 'runner'), self.consume(config, 'wandb', {}, expand=True)
if self.consume(slurmC, 'pre_validate', True): if self.consume(slurmC, 'pre_validate', True):
Runner = self.runners[runnerName] Runner = self.runners[runnerName]
runner = Runner(self, config) runner = Runner(self, config)
@ -296,6 +296,7 @@ class Slate():
) as run: ) as run:
config = copy.deepcopy(orig_config) config = copy.deepcopy(orig_config)
self.deep_update(config, wandb.config) self.deep_update(config, wandb.config)
run.config = copy.deepcopy(orig_config)
runner = Runner(self, config) runner = Runner(self, config)
runner.setup() runner.setup()
runner.run(run) runner.run(run)