sbatch

Author: rohinarora

arch_data.sbatch

@@ -10,4 +10,4 @@
 #SBATCH --gres=gpu:v100-sxm2:1
 
 source activate simclr1
-python pre_train.py --dataset-name cifar10  --arch resnet50 --comment "_cifar10_resnet50"
+python pre_train.py --dataset-name stl10  --arch resnet50 --comment "_stl10_resnet50"

batch1.sbatch

@@ -1,13 +1,13 @@
 #!/bin/bash
 
 #SBATCH --nodes=1
-#SBATCH -c 8
+#SBATCH -c 1
 #SBATCH --time=24:00:00
-#SBATCH --job-name=batch3
+#SBATCH --job-name=batch1
 #SBATCH --partition=multigpu
-#SBATCH --mem=100Gb
+#SBATCH --mem=50Gb
 #SBATCH --output=xepoch.%j.out
 #SBATCH --gres=gpu:v100-sxm2:1
 
 source activate simclr1
-python pre_train.py -b 64
+python pre_train.py -b 64 --comment "_b64"

batch2.sbatch

@@ -3,7 +3,7 @@
 #SBATCH --nodes=1
 #SBATCH -c 8
 #SBATCH --time=24:00:00
-#SBATCH --job-name=batch3
+#SBATCH --job-name=batch2
 #SBATCH --partition=multigpu
 #SBATCH --mem=100Gb
 #SBATCH --output=xepoch.%j.out

temp1.sbatch

@@ -1,17 +1,17 @@
 #!/bin/bash
 
 #SBATCH --nodes=1
-#SBATCH -c 1
+#SBATCH -c 8
 #SBATCH --time=24:00:00
-#SBATCH --job-name=temp4
+#SBATCH --job-name=temp1
 #SBATCH --partition=multigpu
 #SBATCH --mem=100Gb
 #SBATCH --output=xepoch.%j.out
 #SBATCH --gres=gpu:v100-sxm2:1
 
 source activate simclr1
-python pre_train.py --temperature 0.035 &
+python pre_train.py --temperature 0.035 --comment "_tmp35" &
 sleep 60
-python pre_train.py --temperature 0.105 &
+python pre_train.py --temperature 0.105 --comment "_tmp105" &
 sleep 60
-python pre_train.py --temperature 0.14
+python pre_train.py --temperature 0.14 --comment "_tmp14"