Update metrics

2021-06-11 07:20:06 -07:00
parent 248686820c
commit 437f62b055
5 changed files with 158 additions and 1 deletions
--- a/scripts/experimental/train-vit.sh
+++ b/scripts/experimental/train-vit.sh
@@ -28,5 +28,6 @@ python ./exps/basic/xmain.py --save_dir ${save_dir} --rand_seed ${rseed} \
 	--model_config ./configs/yaml.model/vit-cifar10.s0 \
 	--optim_config ./configs/yaml.opt/vit.cifar \
 	--loss_config ./configs/yaml.loss/cross-entropy \
+	--metric_config ./configs/yaml.loss/top-ce \
 	--batch_size 256 \
 	--lr 0.003 --weight_decay 0.3 --scheduler warm-cos --steps 10000