From 81597cdd0a55140f50b32b69507bfa5309b75f44 Mon Sep 17 00:00:00 2001
From: Jordan Gong <jordan.gong@protonmail.com>
Date: Wed, 10 Aug 2022 18:04:50 +0800
Subject: Update config files

---
 supervised/config-vit.yaml | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

(limited to 'supervised/config-vit.yaml')

diff --git a/supervised/config-vit.yaml b/supervised/config-vit.yaml
index 5ed6689..1b1ded7 100644
--- a/supervised/config-vit.yaml
+++ b/supervised/config-vit.yaml
@@ -1,6 +1,6 @@
-codename: cifar10-vit-adam-warmup-anneal
+codename: cifar10-vit-128-adam-warmup-anneal
 seed: -1
-num_iters: 1000
+num_iters: 200
 log_dir: logs
 checkpoint_dir: checkpoints
 
@@ -15,7 +15,7 @@ crop_scale_range:
   - 1
 hflip_prob: 0.5
 
-batch_size: 256
+batch_size: 128
 num_workers: 2
 
 optim: adam
-- 
cgit v1.2.3