do not use max_training_time here

molmod · Jul 23, 2024 · 382f744 · 382f744
1 parent 32da9b2
commit 382f744
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/lumi.yaml b/examples/lumi.yaml
@@ -22,13 +22,13 @@ ModelEvaluation:
     account: "project_465001125"
     nodes_per_block: 1
     cores_per_node: 56
+    max_blocks: 5
     walltime: "01:00:00"
     scheduler_options: "#SBATCH --gres=gpu:8\n"
     worker_init: "ml LUMI/23.09 && ml partition/G && ml rocm/5.6\n"
 ModelTraining:
   cores_per_worker: 7
   gpu: true
-  max_training_time: 5
   slurm:
     partition: "standard-g"
     account: "project_465001125"