14 lines
368 B
Plaintext
14 lines
368 B
Plaintext
|
{
|
||
|
|
"scheduler": ["str", "cos"],
|
||
|
|
"eta_min" : ["float", "0.0"],
|
||
|
|
"epochs" : ["int", "120"],
|
||
|
|
"warmup" : ["int", "5"],
|
||
|
|
"optim" : ["str", "SGD"],
|
||
|
|
"LR" : ["float", "0.1"],
|
||
|
|
"decay" : ["float", "0.0001"],
|
||
|
|
"momentum" : ["float", "0.9"],
|
||
|
|
"nesterov" : ["bool", "1"],
|
||
|
|
"criterion": ["str", "Softmax"],
|
||
|
|
"auxiliary": ["float", "-1"]
|
||
|
|
}
|