update config

mldrugdiscovery · Nov 17, 2020 · dcf5585 · dcf5585
1 parent d782d35
commit dcf5585
Show file tree

Hide file tree

Showing 6 changed files with 4 additions and 8 deletions.
diff --git a/.DS_Store b/.DS_Store
diff --git a/tensorflow/.DS_Store b/tensorflow/.DS_Store
diff --git a/tensorflow/script/.DS_Store b/tensorflow/script/.DS_Store
diff --git a/tensorflow/script/config.py b/tensorflow/script/config.py
@@ -70,7 +70,7 @@
 _C.MODEL = CN()
 _C.MODEL.name            = ''         # The name of the model
 _C.MODEL.depth           = 5          # The input octree depth
-_C.MODEL.depth_out       = 5          # The output feature depth
+_C.MODEL.depth_out       = 2          # The output feature depth
 _C.MODEL.channel         = 3          # The input feature channel
 _C.MODEL.factor          = 1          # The factor used to widen the network
 _C.MODEL.nout            = 40         # The output feature channel

diff --git a/tensorflow/script/configs/test_resnet_depth6.yaml b/tensorflow/script/configs/test_resnet_depth6.yaml
@@ -10,7 +10,6 @@ DATA:
     dtype: points
     distort: True
     depth: 6
-    full_depth: 2
     axis: xyz
     angle: (180, 180, 180)
     interval: (1, 1, 1)
@@ -26,9 +25,9 @@ MODEL:
   channel: 24
   nout: 1
   depth: 6
-  resblock_num: 3
   dropout: (0.0, )
+  resblock_num: 3
 
 LOSS:
   num_class: 1
-  weight_decay: 0.001
+  weight_decay: 0.01
diff --git a/tensorflow/script/configs/train_resnet_depth6.yaml b/tensorflow/script/configs/train_resnet_depth6.yaml
@@ -14,7 +14,6 @@ DATA:
     dtype: points
     distort: True
     depth: 6
-    full_depth: 2
     axis: xyz
     angle: (180, 180, 180)
     interval: (1, 1, 1)
@@ -29,7 +28,6 @@ DATA:
     dtype: points
     distort: True
     depth: 6
-    full_depth: 2
     axis: xyz
     angle: (12, 12, 12)
     interval: (1, 1, 1)
@@ -41,11 +39,10 @@ DATA:
     x_alias: data
 
 MODEL:
-  name: resnet #vgg or resnet
+  name: resnet #resnet or vgg
   channel: 24
   nout: 1
   depth: 6
-  depth_out: 2
   dropout: (0.0,)
   resblock_num: 3