From e3c5a229c30a7a9beb78c68225d0c6d14317c587 Mon Sep 17 00:00:00 2001
From: "ys1087@partner.kit.edu" <ys1087@hkn1990.localdomain>
Date: Thu, 24 Jul 2025 01:10:48 +0200
Subject: [PATCH] require h100 due to gpu ram usage

---
 submit_experiment_batch.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/submit_experiment_batch.py b/submit_experiment_batch.py
index 3a1c92c..64bf3a5 100755
--- 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