VSAG/vision_sjtu_19
3
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

模型更改

Browse Source
This commit is contained in:
xinyang
2019-08-15 16:39:03 +08:00
parent 7a05018cd1
commit 1163cf4c02
2 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tools/TrainCNN/forward.py
View File

@@ -29,19 +29,19 @@ def max_pool_2x2(x):
CONV1_KERNAL_SIZE = 5
# 第一层卷积输出通道数
CONV1_OUTPUT_CHANNELS = 4
CONV1_OUTPUT_CHANNELS = 6
# 第二层卷积核大小
CONV2_KERNAL_SIZE = 3
# 第二层卷积输出通道数
CONV2_OUTPUT_CHANNELS = 8
CONV2_OUTPUT_CHANNELS = 10
# 第三层卷积核大小
CONV3_KERNAL_SIZE = 3
# 第三层卷积输出通道数
CONV3_OUTPUT_CHANNELS = 12
CONV3_OUTPUT_CHANNELS = 14
# 第一层全连接宽度
FC1_OUTPUT_NODES = 60