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

更新CNN参数

Browse Source
This commit is contained in:
xinyang
2019-05-05 18:37:30 +08:00
parent 479e43d2af
commit c09f0d3a2d
10 changed files with 19911 additions and 12337 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
tools/TrainCNN/forward.py
View File

@@ -29,20 +29,19 @@ def max_pool_2x2(x):
CONV1_KERNAL_SIZE = 5
# 第一层卷积输出通道数
CONV1_OUTPUT_CHANNELS = 6
CONV1_OUTPUT_CHANNELS = 8
# 第二层卷积核大小
CONV2_KERNAL_SIZE = 3
# 第二层卷积输出通道数
CONV2_OUTPUT_CHANNELS = 10
CONV2_OUTPUT_CHANNELS = 16
# 第一层全连接宽度
FC1_OUTPUT_NODES = 16
# 第二层全连接宽度（输出标签类型数）
FC2_OUTPUT_NODES = 8
FC2_OUTPUT_NODES = 11
# 输出标签类型数
OUTPUT_NODES = FC2_OUTPUT_NODES