swap-nas/AutoDL-Projects/configs/nas-benchmark/algos/ENAS.config
2024-08-25 18:02:31 +02:00

18 lines
533 B
Plaintext

{
"scheduler": ["str", "cos"],
"LR" : ["float", "0.05"],
"eta_min" : ["float", "0.0005"],
"epochs" : ["int", "250"],
"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"]
}