iicd/xautodl
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

support first-order DARTS on the NASNet search space

Browse Source
This commit is contained in:
D-X-Y
2020-01-17 22:14:47 +11:00
parent 56f2161a3f
commit db2760c260
10 changed files with 213 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
configs/search-archs/DARTS-NASNet-CIFAR.config Normal file
View File

@@ -0,0 +1,9 @@
{
"super_type" : ["str", "nasnet-super"],
"name" : ["str", "GDAS"],
"C" : ["int", "16" ],
"N" : ["int", "2" ],
"steps" : ["int", "4" ],
"multiplier" : ["int", "4" ],
"stem_multiplier" : ["int", "3" ]
}

13
configs/search-opts/DARTS-NASNet-CIFAR.config Normal file
View File

@@ -0,0 +1,13 @@
{
"scheduler": ["str", "cos"],
"LR" : ["float", "0.025"],
"eta_min" : ["float", "0.001"],
"epochs" : ["int", "50"],
"warmup" : ["int", "0"],
"optim" : ["str", "SGD"],
"decay" : ["float", "0.0005"],
"momentum" : ["float", "0.9"],
"nesterov" : ["bool", "1"],
"criterion": ["str", "Softmax"],
"batch_size": ["int", "256"]
}