Add more algorithms

2019-09-28 18:24:47 +10:00
parent bfd6b648fd
commit cfb462e463
286 changed files with 10557 additions and 122955 deletions
--- a/configs/opts/ImageNet-E120-Step-Soft.config
+++ b/configs/opts/ImageNet-E120-Step-Soft.config
@@ -0,0 +1,14 @@
+{
+  "scheduler": ["str",   "multistep"],
+  "epochs"   : ["int",   "120"],
+  "warmup"   : ["int",   "5"],
+  "optim"    : ["str",   "SGD"],
+  "LR"       : ["float", "0.1"],
+  "milestones":["int",   [30, 60, 90]],
+  "gammas"   : ["float", [0.1, 0.1, 0.1]],
+  "decay"    : ["float", "0.0001"],
+  "momentum" : ["float", "0.9"],
+  "nesterov" : ["bool",  "1"],
+  "criterion": ["str",   "Softmax"],
+  "auxiliary": ["float", "-1"]
+}