add yml

L1aoXingyu · Nov 23, 2018 · b297a50 · b297a50
1 parent e229687
commit b297a50
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/configs/train_mnist_softmax.yml b/configs/train_mnist_softmax.yml
@@ -0,0 +1,10 @@
+
+SOLVER:
+  BASE_LR: 0.01
+  WEIGHT_DECAY: 0.0001
+  IMS_PER_BATCH: 128
+
+TEST:
+  IMS_PER_BATCH: 256
+
+OUTPUT_DIR: "/mnt/truenas/scratch/xingyu.liao/checkpoints/mnist_template"