From 2ad81707f8f64e5e2f2896ab5a81fa7c1d15d19c Mon Sep 17 00:00:00 2001
From: D-X-Y <280835372@qq.com>
Date: Sat, 7 Dec 2019 21:39:46 +1100
Subject: [PATCH] add configs for the searched shape (NeurIPS19)

---
 README.md                                 |  3 +++
 configs/NeurIPS-2019/C010-ResNet20.config | 13 +++++++++++++
 configs/NeurIPS-2019/C010-ResNet32.config | 13 +++++++++++++
 configs/NeurIPS-2019/C100-ResNet20.config | 13 +++++++++++++
 configs/NeurIPS-2019/C100-ResNet32.config | 11 +++++++++++
 5 files changed, 53 insertions(+)
 create mode 100644 configs/NeurIPS-2019/C010-ResNet20.config
 create mode 100644 configs/NeurIPS-2019/C010-ResNet32.config
 create mode 100644 configs/NeurIPS-2019/C100-ResNet20.config
 create mode 100644 configs/NeurIPS-2019/C100-ResNet32.config

diff --git a/README.md b/README.md
index bc9348d..8c87cb2 100644
--- a/README.md
+++ b/README.md
@@ -57,6 +57,9 @@ CUDA_VISIBLE_DEVICES=0,1 bash ./scripts-search/search-cifar.sh cifar10 ResNet56
 
 args: `cifar10` indicates the dataset name, `ResNet56` indicates the basemodel name, `CIFARX` indicates the searching hyper-parameters, `0.47/0.57` indicates the expected FLOP ratio, `-1` indicates the random seed.
 
+### Model Configuration
+The searched shapes for ResNet-20/32/56/110/164 in Table 3 in the original paper are listed in [`configs/NeurIPS-2019`](https://github.com/D-X-Y/NAS-Projects/tree/master/configs/NeurIPS-2019).
+
 
 ## [One-Shot Neural Architecture Search via Self-Evaluated Template Network](https://arxiv.org/abs/1910.05733)
 
diff --git a/configs/NeurIPS-2019/C010-ResNet20.config b/configs/NeurIPS-2019/C010-ResNet20.config
new file mode 100644
index 0000000..c22368d
--- /dev/null
+++ b/configs/NeurIPS-2019/C010-ResNet20.config
@@ -0,0 +1,13 @@
+{
+  "dataset"            : ["str"   , "cifar"],
+  "arch"               : ["str"   , "resnet"],
+  "depth"              : ["int"   , "20"],
+  "module"             : ["str"   , "ResNetBasicblock"],
+  "super_type"         : ["str"   , "infer-shape"],
+  "zero_init_residual" : ["bool"  , "0"],
+  "class_num"          : ["int"   , "10"],
+  "search_mode"        : ["str"   , "shape"],
+  "xchannels"          : ["int"   , ["3", "16", "6", "4", "4", "4", "4", "4", "32", "32", "12", "19", "32", "28", "64", "64", "64", "64", "64", "44"]],
+  "xblocks"            : ["int"   , ["3", "3", "3"]],
+  "estimated_FLOP"     : ["float" , "22.444472"]
+}
\ No newline at end of file
diff --git a/configs/NeurIPS-2019/C010-ResNet32.config b/configs/NeurIPS-2019/C010-ResNet32.config
new file mode 100644
index 0000000..dbdef77
--- /dev/null
+++ b/configs/NeurIPS-2019/C010-ResNet32.config
@@ -0,0 +1,13 @@
+{
+  "dataset"            : ["str"   , "cifar"],
+  "arch"               : ["str"   , "resnet"],
+  "depth"              : ["int"   , "32"],
+  "module"             : ["str"   , "ResNetBasicblock"],
+  "super_type"         : ["str"   , "infer-shape"],
+  "zero_init_residual" : ["bool"  , "0"],
+  "class_num"          : ["int"   , "10"],
+  "search_mode"        : ["str"   , "shape"],
+  "xchannels"          : ["int"   , ["3", "16", "6", "4", "12", "4", "4", "16", "9", "9", "6", "14", "32", "32", "9", "19", "28", "9", "32", "19", "32", "9", "64", "64", "64", "64", "64", "64", "64", "32", "38", "32"]],
+  "xblocks"            : ["int"   , ["5", "5", "5"]],
+  "estimated_FLOP"     : ["float" , "34.945344"]
+}
\ No newline at end of file
diff --git a/configs/NeurIPS-2019/C100-ResNet20.config b/configs/NeurIPS-2019/C100-ResNet20.config
new file mode 100644
index 0000000..080424e
--- /dev/null
+++ b/configs/NeurIPS-2019/C100-ResNet20.config
@@ -0,0 +1,13 @@
+{
+  "dataset"            : ["str"   , "cifar"],
+  "arch"               : ["str"   , "resnet"],
+  "depth"              : ["int"   , "20"],
+  "module"             : ["str"   , "ResNetBasicblock"],
+  "super_type"         : ["str"   , "infer-shape"],
+  "zero_init_residual" : ["bool"  , "0"],
+  "class_num"          : ["int"   , "100"],
+  "search_mode"        : ["str"   , "shape"],
+  "xchannels"          : ["int"   , ["3", "16", "4", "4", "4", "4", "6", "4", "32", "32", "9", "19", "32", "28", "64", "64", "64", "64", "64", "64"]],
+  "xblocks"            : ["int"   , ["3", "3", "3"]],
+  "estimated_FLOP"     : ["float" , "22.433792"]
+}
\ No newline at end of file
diff --git a/configs/NeurIPS-2019/C100-ResNet32.config b/configs/NeurIPS-2019/C100-ResNet32.config
new file mode 100644
index 0000000..eb9ac6a
--- /dev/null
+++ b/configs/NeurIPS-2019/C100-ResNet32.config
@@ -0,0 +1,11 @@
+{
+  "dataset"            : ["str"   , "cifar"],
+  "arch"               : ["str"   , "resnet"],
+  "depth"              : ["int"   , "32"],
+  "module"             : ["str"   , "ResNetBasicblock"],
+  "super_type"         : ["str"   , "infer"],
+  "zero_init_residual" : ["bool"  , "0"],
+  "class_num"          : ["int"   , "100"],
+  "xchannels"          : ["int"   , ["3", "16", "4", "4", "4", "14", "6", "4", "8", "4", "4", "4", "32", "32", "9", "28", "28", "28", "28", "28", "32", "32", "64", "64", "64", "64", "64", "64", "64", "64", "64", "64"]],
+  "estimated_FLOP"     : ["float" , "42.493184"]
+}
\ No newline at end of file