2019-11-09 01:36:31 +11:00
|
|
|
{
|
|
|
|
|
"scheduler": ["str", "cos"],
|
|
|
|
|
"LR" : ["float", "0.05"],
|
|
|
|
|
"eta_min" : ["float", "0.0005"],
|
2019-12-24 17:36:47 +11:00
|
|
|
"epochs" : ["int", "250"],
|
2019-11-09 01:36:31 +11:00
|
|
|
"T_max" : ["int", "10"],
|
|
|
|
|
"warmup" : ["int", "0"],
|
|
|
|
|
"optim" : ["str", "SGD"],
|
|
|
|
|
"decay" : ["float", "0.00025"],
|
|
|
|
|
"momentum" : ["float", "0.9"],
|
|
|
|
|
"nesterov" : ["bool", "1"],
|
|
|
|
|
"controller_lr" : ["float", "0.001"],
|
|
|
|
|
"controller_betas": ["float", [0, 0.999]],
|
|
|
|
|
"controller_eps" : ["float", 0.001],
|
|
|
|
|
"criterion": ["str", "Softmax"],
|
|
|
|
|
"batch_size": ["int", "128"]
|
|
|
|
|
}
|
