diff --git a/.DS_Store b/.DS_Store
index 62343de..984f6be 100644
Binary files a/.DS_Store and b/.DS_Store differ
diff --git a/tensorflow/.DS_Store b/tensorflow/.DS_Store
index 2846d94..c278a5c 100644
Binary files a/tensorflow/.DS_Store and b/tensorflow/.DS_Store differ
diff --git a/tensorflow/script/.DS_Store b/tensorflow/script/.DS_Store
index c7139c2..c888a4a 100644
Binary files a/tensorflow/script/.DS_Store and b/tensorflow/script/.DS_Store differ
diff --git a/tensorflow/script/config.py b/tensorflow/script/config.py
index a1acf5f..9418093 100644
--- 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
index e90ee1d..1140474 100644
--- 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
\ No newline at end of file
+  weight_decay: 0.01
\ No newline at end of file
diff --git a/tensorflow/script/configs/train_resnet_depth6.yaml b/tensorflow/script/configs/train_resnet_depth6.yaml
index 6741ebc..bfd303c 100644
--- 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