Prototype generic nas model (cont.).

2020-07-19 08:11:29 +00:00
parent 7ca2ca70b4
commit c34620ab1b
3 changed files with 91 additions and 6 deletions
--- a/configs/nas-benchmark/algos/weight-sharing.config
+++ b/configs/nas-benchmark/algos/weight-sharing.config
@@ -2,7 +2,7 @@
  "scheduler": ["str",   "cos"],
  "LR"       : ["float", "0.025"],
  "eta_min"  : ["float", "0.001"],
-  "epochs"   : ["int",   "250"],
+  "epochs"   : ["int",   "150"],
  "warmup"   : ["int",   "0"],
  "optim"    : ["str",   "SGD"],
  "decay"    : ["float", "0.0005"],