Updated pretrain script

scripts/slurm/run_glueTOne.sh

@@ -3,16 +3,16 @@
 #SBATCH --job-name=galore_pretrain   # Name des Jobs
 #SBATCH --nodes=1                    # 1 Knoten nutzen
 #SBATCH --ntasks=1                    # 1 Aufgabe
-#SBATCH --mem=32G                     # 32 GB RAM zuweisen
-#SBATCH --cpus-per-task=4             # 4 CPU-Kerne pro Task
+#SBATCH --mem=16G                     # 32 GB RAM zuweisen
+#SBATCH --cpus-per-task=2             # 4 CPU-Kerne pro Task
 #SBATCH --gres=gpu:1                  # 1 GPU anfordern
-#SBATCH --time=04:00:00               # Maximale Laufzeit von 1 Stunde
+#SBATCH --time=01:00:00               # Maximale Laufzeit von 1 Stunde
 #SBATCH --nodelist=workg02
 
 # Docker-Image von Docker Hub ausführen
 srun --container-image=docker://mcr.informatik.uni-halle.de#apcne/galore-replication \
      --container-mounts="$1":/workspace \
-     --container-name=ml_gitlab2 \
+     --container-name=galore_1 \
      --container-remap-root \
      --container-writable \
      --gres=gpu:1 \

scripts/slurm/run_pretrain.sh

@@ -3,17 +3,19 @@
 #SBATCH --job-name=galore_pretrain   # Name des Jobs
 #SBATCH --nodes=1                    # 1 Knoten nutzen
 #SBATCH --ntasks=1                    # 1 Aufgabe
-#SBATCH --mem=32G                     # 32 GB RAM zuweisen
-#SBATCH --cpus-per-task=4             # 4 CPU-Kerne pro Task
+#SBATCH --mem=16G                     # 32 GB RAM zuweisen
+#SBATCH --cpus-per-task=2             # 4 CPU-Kerne pro Task
 #SBATCH --gres=gpu:1                  # 1 GPU anfordern
-#SBATCH --time=00:05:00               # Maximale Laufzeit von 5 Minuten
-
-# Set TMPDIR to a different directory
-export TMPDIR=/home/apzgb/tmp
+#SBATCH --time=00:05:00               # Maximale Laufzeit von 1 Stunde
+#SBATCH --nodelist=workg02
 
 # Docker-Image von Docker Hub ausführen
-srun \
-    --container-image=docker://tommotius/galore_image \
-    --container-name=ml-container \
-    --container-mounts=/home/apzgb/Dokumente/GaLoreReplication:/workspace \
-    bash -i -c "cd /workspace && chmod +x ./scripts/shell/pretrain.sh"
\ No newline at end of file
+srun --container-image=docker://mcr.informatik.uni-halle.de#apcne/galore-replication \
+     --container-mounts="$1":/workspace \
+     --container-name=galore_1 \
+     --container-remap-root \
+     --container-writable \
+     --gres=gpu:1 \
+     --cpus-per-gpu=2 \
+     --mem-per-cpu=16G \
+     --pty bash -c "cd /workspace && bash scripts/shell/pretrain_60m.sh"