require h100 due to gpu ram usage

2025-07-24 01:10:48 +02:00 · 2025-07-24 01:10:48 +02:00 · e3c5a229c3
commit e3c5a229c3
parent 69502c8911
1 changed files with 1 additions and 1 deletions
--- a/submit_experiment_batch.py
+++ b/submit_experiment_batch.py
@ -69,7 +69,7 @@ def create_job_script(task, config, seed, phase):
    script_content = f'''#!/bin/bash
 #SBATCH --job-name=fasttd3_{phase}_{task.replace("-", "_")}_s{seed}
 #SBATCH --account=hk-project-p0022232
-#SBATCH --partition=accelerated
+#SBATCH --partition=accelerated-h100
 #SBATCH --time={config["time_limit"]}
 #SBATCH --gres=gpu:1
 #SBATCH --ntasks=1