diff --git a/examples/simple/submit_job.py b/examples/simple/submit_job.py index a884a7c..3ec1ed2 100644 --- a/examples/simple/submit_job.py +++ b/examples/simple/submit_job.py @@ -1,5 +1,10 @@ +import yaml + import easy_slurm +with open("assets/hparams.yaml") as f: + config = yaml.safe_load(f) + easy_slurm.submit_job( job_dir="$HOME/.local/share/easy_slurm/{date}-{job_name}", src=["./src", "./assets"], @@ -23,7 +28,7 @@ # Do any cleanup tasks here. """, sbatch_options={ - "job-name": "example-simple", + "job-name": "example-simple-epochs={hp.epochs}", "account": "def-ibajic", "time": "0:03:00", "nodes": "1", @@ -31,4 +36,5 @@ "cpus-per-task": "1", "mem": "4000M", }, + config=config, ) diff --git a/examples/simple_yaml/job.yaml b/examples/simple_yaml/job.yaml index 9fe4fec..9a4eeb1 100644 --- a/examples/simple_yaml/job.yaml +++ b/examples/simple_yaml/job.yaml @@ -17,10 +17,11 @@ on_run_resume: "cd src && python main.py --resume" teardown: | # Do any cleanup tasks here. sbatch_options: - job-name: "example-simple" + job-name: "example-simple-epochs={hp.epochs}" account: "def-ibajic" time: "0:03:00" nodes: 1 ntasks-per-node: 1 cpus-per-task: 1 mem: "4000M" +# config: {} # Loaded from assets/hparams.yaml. diff --git a/examples/simple_yaml/submit_job.py b/examples/simple_yaml/submit_job.py index ee78a32..9d373b1 100644 --- a/examples/simple_yaml/submit_job.py +++ b/examples/simple_yaml/submit_job.py @@ -7,7 +7,11 @@ def main(): with open("job.yaml") as f: job_config = yaml.safe_load(f) - easy_slurm.submit_job(**job_config) + # Optional: Load config for easy job name formatting. + with open("assets/hparams.yaml") as f: + config = yaml.safe_load(f) + + easy_slurm.submit_job(**job_config, config=config) if __name__ == "__main__":