减少CNN训练时，测试集的内存占用

2019-05-03 10:50:10 +08:00
parent 86e04d444c
commit 174809e6a7
3 changed files with 18 additions and 13 deletions
--- a/tools/TrainCNN/forward.py
+++ b/tools/TrainCNN/forward.py
@@ -41,7 +41,7 @@ CONV2_OUTPUT_CHANNELS = 10
 FC1_OUTPUT_NODES = 16

 # 第二层全连接宽度（输出标签类型数）
-FC2_OUTPUT_NODES = 4
+FC2_OUTPUT_NODES = 8

 # 输出标签类型数
 OUTPUT_NODES = FC2_OUTPUT_NODES