aboutsummaryrefslogtreecommitdiff
path: root/supervised/config.yaml
diff options
context:
space:
mode:
authorJordan Gong <jordan.gong@protonmail.com>2022-08-10 17:02:27 +0800
committerJordan Gong <jordan.gong@protonmail.com>2022-08-10 17:02:27 +0800
commit95d5e4e82df0de08210088d07d6d89692f1325d1 (patch)
treeed6b6788383ba77ff439afdc6da4490cb467eba9 /supervised/config.yaml
parenta035bbdd0cc5e39954272d13d1e1595db738846a (diff)
Add supervised ViT baseline
Diffstat (limited to 'supervised/config.yaml')
-rw-r--r--supervised/config.yaml3
1 files changed, 3 insertions, 0 deletions
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