update slurms.

Author: v0lta

scripts/integrated_gradients.slurm

@@ -2,17 +2,15 @@
 #
 #SBATCH --nodes=1
 #SBATCH --tasks-per-node=1
-#SBATCH --partition=A40short
+#SBATCH --partition=A100devel
 #SBATCH --job-name=tr_net
 #SBATCH --output=./out/train_net-%j.out
 #SBATCH --error=./out/train_net-%j.err
 #SBATCH --gres gpu:1
 #SBATCH --cpus-per-task=8
 echo "training network..."
 
-module load cuDNN
-module load Python
-pip install "jax[cuda11_cudnn82]" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
+module load Anaconda3
 pip install -r requirements.txt
 
 python ./src/mnist_integrated.py

scripts/train.slurm

@@ -2,17 +2,15 @@
 #
 #SBATCH --nodes=1
 #SBATCH --tasks-per-node=1
-#SBATCH --partition=A40short
+#SBATCH --partition=A40devel
 #SBATCH --job-name=tr_net
 #SBATCH --output=./out/train_net-%j.out
 #SBATCH --error=./out/train_net-%j.err
 #SBATCH --gres gpu:1
 #SBATCH --cpus-per-task=8
 echo "training network..."
 
-module load cuDNN
-module load Python
-pip install "jax[cuda11_cudnn82]" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
+module load Anaconda3
 pip install -r requirements.txt
 
 python ./src/deepfake_interpretation.py