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