add mobilenetv1 yml

itisianlee · Apr 5, 2021 · 0240e34 · 0240e34
1 parent b1d48dc
commit 0240e34
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/yamls/mobilenetv1.yml b/yamls/mobilenetv1.yml
@@ -0,0 +1,18 @@
+CLASSIFIER: mobilenet_v1
+YAML: common.yml
+COMMON:
+  workers: 4
+  batch_size: 128
+  weight_decay: 5e-4
+  epochs: 300
+  data_path: data/cifar-100-python.tar.gz
+  logdir: logs
+  verbose: 1
+  seed: 2021
+OPTIMIZER:
+  name: momentum
+  params:
+    momentum: 0.9
+SCHEDULER:
+  name: cosine
+  warm_up_step: 2000