From 47c0b7b2c85b0afa9895ed77a8135b80f3ad83be Mon Sep 17 00:00:00 2001 From: Dominik Roth Date: Wed, 12 Jul 2023 18:02:10 +0200 Subject: [PATCH] maybe fix wandb sweeps --- slate/slate.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/slate/slate.py b/slate/slate.py index a73c0d8..2620b04 100644 --- a/slate/slate.py +++ b/slate/slate.py @@ -183,8 +183,8 @@ class Slate(): schedC = self.consume(config, 'scheduler') s_name = self.consume(slurmC, 'name') + # Pre Validation runnerName, _ = self.consume(config, 'runner'), self.consume(config, 'wandb', {}, expand=True) - if self.consume(slurmC, 'pre_validate', True): Runner = self.runners[runnerName] runner = Runner(self, config) @@ -296,6 +296,7 @@ class Slate(): ) as run: config = copy.deepcopy(orig_config) self.deep_update(config, wandb.config) + run.config = copy.deepcopy(orig_config) runner = Runner(self, config) runner.setup() runner.run(run)