iicd/autodl-projects
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update yaml configs

Browse Source
This commit is contained in:
D-X-Y
2021-06-10 21:53:22 +08:00
parent 1a7440d2af
commit 9bf0fa5f04
21 changed files with 410 additions and 178 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
scripts/experimental/train-vit.sh
View File

@@ -22,6 +22,10 @@ save_dir=./outputs/${dataset}/vit-experimental
python --version
python ./exps/basic/xmain.py --save_dir ${save_dir} --rand_seed ${rseed} \
--train_data_config ./configs/data.yaml/${dataset}.train \
--valid_data_config ./configs/data.yaml/${dataset}.test \
--data_path $TORCH_HOME/cifar.python
--train_data_config ./configs/yaml.data/${dataset}.train \
--valid_data_config ./configs/yaml.data/${dataset}.test \
--data_path $TORCH_HOME/cifar.python \
--model_config ./configs/yaml.model/vit-cifar10.s0 \
--optim_config ./configs/yaml.opt/vit.cifar \
--loss_config ./configs/yaml.loss/cross-entropy \
--lr 0.003 --weight_decay 0.3