smart_augmentation/jobs/benchmark.sh
2024-08-20 11:53:35 +02:00

34 lines
No EOL
919 B
Bash

#!/bin/bash
#SBATCH --gres=gpu:v100l:1 # https://docs.computecanada.ca/wiki/Using_GPUs_with_Slurm
#SBATCH --cpus-per-task=6 #6 # Cores proportional to GPUs: 6 on Cedar, 16 on Graham.
#SBATCH --mem=32000M #32000M # Memory proportional to CPUs: 32000 Cedar, 64000 Graham.
#SBATCH --account=def-mpederso
#SBATCH --time=20:00:00
#SBATCH --job-name=Benchmark
#SBATCH --output=log/%x-%j.out
#SBATCH --mail-user=harle.collette.antoine@gmail.com
#SBATCH --mail-type=END
#SBATCH --mail-type=FAIL
#SBATCH --array=1-3
# Setup
source ~/dataug/bin/activate
#Execute
# echo $(pwd) = /home/antoh/projects/def-mpederso/antoh/smart_augmentation/jobs
cd ../higher/smart_aug/
python test_dataug.py \
-a true \
-m resnet18 \
-ep 200 \
-K 1 \
-N 3 \
-tfc ../config/invScale_wide_tf_config2.json \
-ds CIFAR10 \
-al 10 \
-wu 2 \
-rf '../res/benchmark/' \
-pf __$SLURM_ARRAY_TASK_ID