Modify ConvConfig, Add depth dimension

8 years ago · 8cc0eb9c5d
parent cfb86c4e23
commit 8cc0eb9c5d
1 changed files with 7 additions and 1 deletions
--- a/proto/ModelConfig.proto
+++ b/proto/ModelConfig.proto
@ -82,6 +82,12 @@ message ConvConfig {

  // if not set, use img_size
  optional uint32 img_size_y = 14;
+  
+  optional uint32 filter_size_z = 15 [ default = 1 ];
+  optional uint32 padding_z = 16 [ default = 1 ];
+  optional uint32 stride_z = 17 [ default = 1 ];
+  optional uint32 output_z = 18 [ default = 1 ];
+  optional uint32 img_size_z = 19 [ default = 1 ];
 }

 message PoolConfig {