-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathwrn_small_taylor.sh
4 lines (4 loc) · 2.71 KB
/
wrn_small_taylor.sh
1
2
3
4
CUDA_VISIBLE_DEVICES=0 python taylorized_train.py --epochs 200 --init_seed 100 --model wideresnet --wideresnet_bs 2 --wideresnet_k 4 --wideresnet_channels 128 --loss logistic --parameterization standard --linearize --optimizer momentum --lr 0.03162 --momentum 0.0 --weight_decay 0.0 --lr_decay --decay_epoch 100 --decay_epoch_2 150 --decay_factor 0.1 --grad_norm_thresh 10 --batch_size_train 256 --batch_size_test 256 --logdir runs/WRNSMALL-LIN-lr-0.03162-clip-10-bs-256 --save_steps 200 --early_save_steps 25 --early_save_till_step 200 --load_path saved_models/WRNSMALL-lr-0.03162-clip-10-bs-256/0.npy --save_path saved_models/WRNSMALL-LIN-lr-0.03162-clip-10-bs-256 &
CUDA_VISIBLE_DEVICES=1 python taylorized_train.py --epochs 200 --init_seed 100 --model wideresnet --wideresnet_bs 2 --wideresnet_k 4 --wideresnet_channels 128 --loss logistic --parameterization standard --expand_order 2 --optimizer momentum --lr 0.03162 --momentum 0.0 --weight_decay 0.0 --lr_decay --decay_epoch 100 --decay_epoch_2 150 --decay_factor 0.1 --grad_norm_thresh 10 --batch_size_train 256 --batch_size_test 256 --logdir runs/WRNSMALL-QUAD-lr-0.03162-clip-10-bs-256 --save_steps 200 --early_save_steps 25 --early_save_till_step 200 --load_path saved_models/WRNSMALL-lr-0.03162-clip-10-bs-256/0.npy --save_path saved_models/WRNSMALL-QUAD-lr-0.03162-clip-10-bs-256 &
CUDA_VISIBLE_DEVICES=2 python taylorized_train.py --epochs 200 --init_seed 100 --model wideresnet --wideresnet_bs 2 --wideresnet_k 4 --wideresnet_channels 128 --loss logistic --parameterization standard --expand_order 3 --optimizer momentum --lr 0.03162 --momentum 0.0 --weight_decay 0.0 --lr_decay --decay_epoch 100 --decay_epoch_2 150 --decay_factor 0.1 --grad_norm_thresh 10 --batch_size_train 256 --batch_size_test 256 --logdir runs/WRNSMALL-CUBIC-lr-0.03162-clip-10-bs-256 --save_steps 200 --early_save_steps 25 --early_save_till_step 200 --load_path saved_models/WRNSMALL-lr-0.03162-clip-10-bs-256/0.npy --save_path saved_models/WRNSMALL-CUBIC-lr-0.03162-clip-10-bs-256 &
CUDA_VISIBLE_DEVICES=3 python taylorized_train.py --epochs 200 --init_seed 100 --model wideresnet --wideresnet_bs 2 --wideresnet_k 4 --wideresnet_channels 128 --loss logistic --parameterization standard --expand_order 4 --optimizer momentum --lr 0.03162 --momentum 0.0 --weight_decay 0.0 --lr_decay --decay_epoch 100 --decay_epoch_2 150 --decay_factor 0.1 --grad_norm_thresh 10 --batch_size_train 256 --batch_size_test 256 --logdir runs/WRNSMALL-QUARTIC-lr-0.03162-clip-10-bs-256 --save_steps 200 --early_save_steps 25 --early_save_till_step 200 --load_path saved_models/WRNSMALL-lr-0.03162-clip-10-bs-256/0.npy --save_path saved_models/WRNSMALL-QUARTIC-lr-0.03162-clip-10-bs-256