From 95d5e4e82df0de08210088d07d6d89692f1325d1 Mon Sep 17 00:00:00 2001 From: Jordan Gong Date: Wed, 10 Aug 2022 17:02:27 +0800 Subject: Add supervised ViT baseline --- supervised/config.yaml | 3 +++ 1 file changed, 3 insertions(+) (limited to 'supervised/config.yaml') diff --git a/supervised/config.yaml b/supervised/config.yaml index 91d6dc0..a3cf4f6 100644 --- a/supervised/config.yaml +++ b/supervised/config.yaml @@ -4,6 +4,9 @@ num_iters: 1000 log_dir: logs checkpoint_dir: checkpoints +backbone: resnet +label_smooth: 0 + dataset: cifar10 dataset_dir: dataset crop_size: 32 -- cgit v1.2.3