mtb.sh (1406B)
1 #!/bin/bash 2 #SBATCH -A lco@gpu 3 #SBATCH -C v100-32g 4 #SBATCH --job-name=gbure_mtb 5 #SBATCH --ntasks=1 6 #SBATCH --gres=gpu:1 7 #SBATCH --cpus-per-task=10 8 #SBATCH --distribution=block:block 9 #SBATCH --hint=nomultithread 10 #SBATCH --time=20:00:00 11 #SBATCH --output=/gpfswork/rech/lco/url46ht/Étienne/runs/%x_%j.stdout 12 #SBATCH --error=/gpfswork/rech/lco/url46ht/Étienne/runs/%x_%j.stderr 13 #SBATCH --array=0-3 14 15 # %x = nom du job 16 # %j = id du job 17 18 module purge 19 source ~/.bashrc 20 conda activate Étienne 21 export DATA_PATH=$WORK/Étienne/data 22 export LOG_PATH=$WORK/Étienne/log 23 export HF_DATASETS_OFFLINE=1 24 export TRANSFORMERS_OFFLINE=1 25 cd $WORK/Étienne/code 26 27 # echo des commandes lancées 28 set -x 29 30 config="" 31 32 if [ $(( $SLURM_ARRAY_TASK_ID % 2 )) -eq 0 ]; then 33 config="$config --latent_metric_scale=\"standard\" --latent_dot_mean=1067.65 --latent_dot_std=111.17" 34 else 35 config="$config --latent_metric_scale=\"sqrt\" --latent_dot_mean=None --latent_dot_std=None" 36 fi 37 38 if [ $(( $SLURM_ARRAY_TASK_ID / 2 % 2 )) -eq 0 ]; then 39 config="$config --filter_empty_neighborhood=True" 40 else 41 config="$config --filter_empty_neighborhood=False" 42 fi 43 44 config="$config --seed=$(($SLURM_ARRAY_TASK_ID / 4))" 45 46 # We use the gcn_mtb config with a 0 topological weight in order to get accuracies in function of neighborhood sizes 47 python -m gbure.train gbure/config/gcn_mtb.py $config --post_transformer_layer=\"layer_norm\" --topological_weight=0