Add more algorithms

2019-09-28 18:24:47 +10:00
parent bfd6b648fd
commit cfb462e463
286 changed files with 10557 additions and 122955 deletions
--- a/configs/archs/ImageNet-ResNet18V1.config
+++ b/configs/archs/ImageNet-ResNet18V1.config
@@ -0,0 +1,11 @@
+{
+  "dataset"         : ["str",  "imagenet"],
+  "arch"            : ["str",  "resnet"],
+  "block_name"      : ["str",  "BasicBlock"],
+  "layers"          : ["int",  [2,2,2,2]],
+  "deep_stem"       : ["bool", 0],
+  "zero_init_residual" : ["bool",  "1"],
+  "groups"          : ["int",  1],
+  "width_per_group" : ["int", 64],
+  "norm_layer"      : ["none", "None"]
+}