-
Notifications
You must be signed in to change notification settings - Fork 4
/
tti.sh
35 lines (28 loc) · 1.58 KB
/
tti.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
#!/bin/sh
#SBATCH --output=path/test_time_interventions_%j.out
pwd; hostname; date
CURRENT=`date +"%Y-%m-%d_%T"`
echo $CURRENT
slurm_output_cub_vit=cub_vit_$CURRENT.out
slurm_output_cub_resnet=cub_resnet_$CURRENT.out
slurm_output_awa2_vit=awa2_vit_$CURRENT.out
slurm_output_awa2_resnet=awa2_resnet_$CURRENT.out
echo "Test Time Interventions"
source path-of-conda/anaconda3/etc/profile.d/conda.sh
conda activate python_3_7_rtx_6000
# -----------------------------------------------------
# CUB_VIT
# -----------------------------------------------------
python ../codebase/test_time_interventions_main.py --model "MoIE" --arch "ViT-B_16" --dataset "cub" --iterations 6 --top_K 3 5 10 15 20 25 30 50 > $slurm_output_cub_vit
# -----------------------------------------------------
# CUB_ResNet
# -----------------------------------------------------
python ../codebase/test_time_interventions_main.py --model "MoIE" --arch "ResNet101" --dataset "cub" --iterations 6 --top_K 3 5 10 15 20 25 30 50 > $slurm_output_cub_resnet
# -----------------------------------------------------
# Awa2_vit
# -----------------------------------------------------
python ../codebase/test_time_interventions_main.py --model "MoIE" --arch "ViT-B_16" --dataset "awa2" --iterations 6 --top_K 3 5 10 15 20 25 30 50 > $slurm_output_awa2_vit
# -----------------------------------------------------
# Awa2_ResNet
# -----------------------------------------------------
python ../codebase/test_time_interventions_main.py --model "MoIE" --arch "ResNet101" --dataset "awa2" --iterations 4 --top_K 3 5 10 15 20 25 30 50 > $slurm_output_awa2_resnet