Mathis/CUDANet
1
0
Fork 0
mirror of https://github.com/lordmathis/CUDANet.git synced 2025-11-06 01:34:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix branch names

Browse Source
This commit is contained in:
Matúš Námešný
2024-05-30 15:21:01 +02:00
parent ac53b1672e
commit 9faf20876a
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
examples/inception_v3/inception_v3.cpp
View File

@@ -219,7 +219,7 @@ class InceptionB : public CUDANet::Module {
// Branch 3x3 // Branch 3x3
branch3x3 = new BasicConv2d( branch3x3 = new BasicConv2d(
inputShape, inputChannels, 384, {3, 3}, {2, 2}, {0, 0}, inputShape, inputChannels, 384, {3, 3}, {2, 2}, {0, 0},
prefix + ".branch1x1" prefix + ".branch3x3"
); );
addLayer("", branch3x3); addLayer("", branch3x3);
@@ -230,7 +230,7 @@ class InceptionB : public CUDANet::Module {
); );
addLayer("", branch3x3dbl_1); addLayer("", branch3x3dbl_1);
branch3x3dbl_2 = new BasicConv2d( branch3x3dbl_2 = new BasicConv2d(
branch3x3dbl_1->getOutputDims(), 96, 96, {3, 3}, {1, 1}, {1, 1}, branch3x3dbl_1->getOutputDims(), 64, 96, {3, 3}, {1, 1}, {1, 1},
prefix + ".branch3x3dbl_2" prefix + ".branch3x3dbl_2"
); );
addLayer("", branch3x3dbl_2); addLayer("", branch3x3dbl_2);
@@ -485,7 +485,7 @@ class InceptionD : public CUDANet::Module {
// Branch 3x3 // Branch 3x3
branch3x3_1 = new BasicConv2d( branch3x3_1 = new BasicConv2d(
inputShape, inputChannels, 192, {1, 1}, {1, 1}, {0, 0}, inputShape, inputChannels, 192, {1, 1}, {1, 1}, {0, 0},
prefix + ".branch3x3" prefix + ".branch3x3_1"
); );
addLayer("", branch3x3_1); addLayer("", branch3x3_1);
branch3x3_2 = new BasicConv2d( branch3x3_2 = new BasicConv2d(