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

better warning about uncosnumed configs

Browse Source
This commit is contained in:
Dominik Moritz Roth 2024-02-16 13:57:48 +01:00
parent 6c29d49a33
commit 65350ade74
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
slate/slate.py
View File

@ -237,7 +237,9 @@ class Slate():
num_parallel_jobs = min(self.consume(slurmC, 'num_parallel_jobs', num_jobs), num_jobs)
array = f'0-{last_job_idx}%{num_parallel_jobs}'
job = pyslurm.JobSubmitDescription(name=s_name, script=script, array=array, **slurmC)
if self.verify:
if config != {}:
pprint('Unconsumed Config: ', config)
if self.verify or config != {}:
input(f'<Press enter to submit {num_jobs} job(s) to slurm>')
job_id = job.submit()
print(f'[>] Job submitted to slurm with id {job_id}')
@ -323,7 +325,7 @@ class Slate():
**wandbC
) as run:
runner = Runner(self, runnerConf)
runner.setup(wandbC['job_type'])
runner.setup(wandbC['group']+wandbC['job_type'])
runner.run(run)
except wandb.errors.CommError as e:
retry -= 1
@ -363,7 +365,7 @@ class Slate():
self.deep_update(config, wandb.config)
run.config = copy.deepcopy(config)
runner = Runner(self, config)
runner.setup(wandbC['job_type'])
runner.setup(wandbC['group']+wandbC['job_type'])
runner.run(run)
if config != {}: