Adding Stage 1 to config.

rameshnair007 · Apr 29, 2020 · bde4a76 · bde4a76
1 parent 91ff90e
commit bde4a76
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/lib/config/models.py b/lib/config/models.py
@@ -17,6 +17,14 @@
 HIGH_RESOLUTION_NET.FINAL_CONV_KERNEL = 1
 HIGH_RESOLUTION_NET.WITH_HEAD = True
 
+HIGH_RESOLUTION_NET.STAGE1 = CN()
+HIGH_RESOLUTION_NET.STAGE1.NUM_MODULES = 1
+HIGH_RESOLUTION_NET.STAGE1.NUM_BRANCHES = 1
+HIGH_RESOLUTION_NET.STAGE1.NUM_BLOCKS = [4]
+HIGH_RESOLUTION_NET.STAGE1.NUM_CHANNELS = [32]
+HIGH_RESOLUTION_NET.STAGE1.BLOCK = 'BOTTLENECK'
+HIGH_RESOLUTION_NET.STAGE1.FUSE_METHOD = 'SUM'
+
 HIGH_RESOLUTION_NET.STAGE2 = CN()
 HIGH_RESOLUTION_NET.STAGE2.NUM_MODULES = 1
 HIGH_RESOLUTION_NET.STAGE2.NUM_BRANCHES = 2