-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathtrain_kld.sh
More file actions
executable file
·25 lines (21 loc) · 2.78 KB
/
Copy pathtrain_kld.sh
File metadata and controls
executable file
·25 lines (21 loc) · 2.78 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
#!/bin/bash
#SBATCH -A mobility_arfs
#SBATCH --partition ihub
#SBATCH -n 15
#SBATCH --gres=gpu:2
#SBATCH --mem-per-cpu=2048
#SBATCH --time=4-00:00:00
#SBATCH --mail-type=END
#SBATCH -w gnode109
python train_step2.py --save_dir '/scratch/kushagra0301/kld_0.9' --num_epochs 5 --dataset_avoided 'NA' --saved_model /scratch/kushagra0301/Crack_IL_step_1/best_model.pth --kld_weight 0.9 > 'kld0.9_step2.log' 2>&1
python train_step2.py --save_dir '/scratch/kushagra0301/kld_0.7' --num_epochs 5 --dataset_avoided 'NA' --saved_model /scratch/kushagra0301/Crack_IL_step_1/best_model.pth --kld_weight 0.7 > 'kld0.7_step2.log' 2>&1
python train_step2.py --save_dir '/scratch/kushagra0301/kld_0.5' --num_epochs 5 --dataset_avoided 'NA' --saved_model /scratch/kushagra0301/Crack_IL_step_1/best_model.pth --kld_weight 0.5 > 'kld0.5_step2.log' 2>&1
python train_step2.py --save_dir '/scratch/kushagra0301/kld_0.3' --num_epochs 5 --dataset_avoided 'NA' --saved_model /scratch/kushagra0301/Crack_IL_step_1/best_model.pth --kld_weight 0.3 > 'kld0.3_step2.log' 2>&1
python train_step2.py --save_dir '/scratch/kushagra0301/kld_0.1' --num_epochs 5 --dataset_avoided 'NA' --saved_model /scratch/kushagra0301/Crack_IL_step_1/best_model.pth --kld_weight 0.1 > 'kld0.1_step2.log' 2>&1
# python train_step2.py --save_dir '/scratch/kushagra0301/ce_0.1' --num_epochs 30 --dataset_avoided 'NA' --saved_model /scratch/kushagra0301/Crack_IL_step_1/best_model.pth --ce_weight 0.1 > 'ce0.1_step2.log' 2>&1
# python train_step2.py --save_dir '/scratch/kushagra0301/ce_0.2' --num_epochs 30 --dataset_avoided 'NA' --saved_model /scratch/kushagra0301/Crack_IL_step_1/best_model.pth --ce_weight 0.2 > 'ce0.2_step2.log' 2>&1
# python train_step2.py --save_dir '/scratch/kushagra0301/ce_0.3' --num_epochs 30 --dataset_avoided 'NA' --saved_model /scratch/kushagra0301/Crack_IL_step_1/best_model.pth --ce_weight 0.3 > 'ce0.3_step2.log' 2>&1
# python train_step2.py --save_dir '/scratch/kushagra0301/ce_0.4' --num_epochs 30 --dataset_avoided 'NA' --saved_model /scratch/kushagra0301/Crack_IL_step_1/best_model.pth --ce_weight 0.4 > 'ce0.4_step2.log' 2>&1
# python train_step2.py --save_dir '/scratch/kushagra0301/ce_0.5' --num_epochs 30 --dataset_avoided 'NA' --saved_model /scratch/kushagra0301/Crack_IL_step_1/best_model.pth --ce_weight 0.5 > 'ce0.5_step2.log' 2>&1
# python train_step2.py --save_dir '/scratch/kushagra0301/Crack_IL_step_2_30' --num_epochs 150 --source_dataset_path '/scratch/kushagra0301/CrackDataset30' --dataset_avoided 'NA' --saved_model /scratch/kushagra0301/Crack_IL_step_1_30/best_model.pth > 'train_step2_30.log' 2>&1
# python train_step2.py --save_dir '/scratch/kushagra0301/source_as_target' --num_epochs 150 --dataset_avoided 'NA' --saved_model /scratch/kushagra0301/source_as_target/best_model.pth > 'source_as_target_step2.log' 2>&1