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

Format inception

Browse Source
This commit is contained in:
Matúš Námešný
2024-05-26 20:06:11 +02:00
parent ca9c4619e9
commit 07d505a0e5
1 changed files with 17 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
examples/inception_v3/inception_v3.cpp
View File

@@ -276,16 +276,16 @@ class InceptionC : public CUDANet::Module {
"branch7x7dbl_1" "branch7x7dbl_1"
); );
branch7x7dbl_2 = new BasicConv2d( branch7x7dbl_2 = new BasicConv2d(
branch7x7dbl_1->getOutputDims(), nChannels_7x7, nChannels_7x7, {7, 1}, branch7x7dbl_1->getOutputDims(), nChannels_7x7, nChannels_7x7,
{1, 1}, {3, 0}, "branch7x7dbl_2" {7, 1}, {1, 1}, {3, 0}, "branch7x7dbl_2"
); );
branch7x7dbl_3 = new BasicConv2d( branch7x7dbl_3 = new BasicConv2d(
branch7x7dbl_2->getOutputDims(), nChannels_7x7, nChannels_7x7, {1, 7}, branch7x7dbl_2->getOutputDims(), nChannels_7x7, nChannels_7x7,
{1, 1}, {0, 3}, "branch7x7dbl_3" {1, 7}, {1, 1}, {0, 3}, "branch7x7dbl_3"
); );
branch7x7dbl_4 = new BasicConv2d( branch7x7dbl_4 = new BasicConv2d(
branch7x7dbl_3->getOutputDims(), nChannels_7x7, nChannels_7x7, {7, 1}, branch7x7dbl_3->getOutputDims(), nChannels_7x7, nChannels_7x7,
{1, 1}, {3, 0}, "branch7x7dbl_4" {7, 1}, {1, 1}, {3, 0}, "branch7x7dbl_4"
); );
branch7x7dbl_5 = new BasicConv2d( branch7x7dbl_5 = new BasicConv2d(
branch7x7dbl_4->getOutputDims(), nChannels_7x7, 192, {1, 7}, {1, 1}, branch7x7dbl_4->getOutputDims(), nChannels_7x7, 192, {1, 7}, {1, 1},
@@ -294,7 +294,8 @@ class InceptionC : public CUDANet::Module {
// Branch Pool // Branch Pool
branchPool_1 = new CUDANet::Layers::AvgPooling2d( branchPool_1 = new CUDANet::Layers::AvgPooling2d(
inputSize, inputChannels, {3, 3}, {1, 1}, {1, 1}, CUDANet::Layers::ActivationType::NONE inputSize, inputChannels, {3, 3}, {1, 1}, {1, 1},
CUDANet::Layers::ActivationType::NONE
); );
branchPool_2 = new BasicConv2d( branchPool_2 = new BasicConv2d(
branchPool_1->getOutputDims(), inputChannels, 192, {1, 1}, {1, 1}, branchPool_1->getOutputDims(), inputChannels, 192, {1, 1}, {1, 1},
@@ -303,21 +304,17 @@ class InceptionC : public CUDANet::Module {
// Concat // Concat
concat_1 = new CUDANet::Layers::Concat( concat_1 = new CUDANet::Layers::Concat(
branch1x1->getOutputSize(), branch1x1->getOutputSize(), branch7x7_3->getOutputSize()
branch7x7_3->getOutputSize()
); );
concat_2 = new CUDANet::Layers::Concat( concat_2 = new CUDANet::Layers::Concat(
concat_1->getOutputSize(), concat_1->getOutputSize(), branch7x7dbl_5->getOutputSize()
branch7x7dbl_5->getOutputSize()
); );
concat_3 = new CUDANet::Layers::Concat( concat_3 = new CUDANet::Layers::Concat(
concat_2->getOutputSize(), concat_2->getOutputSize(), branchPool_2->getOutputSize()
branchPool_2->getOutputSize()
); );
} }
float *forward(const float *d_input) { float *forward(const float *d_input) {
float *branch1x1_output = branch1x1->forward(d_input); float *branch1x1_output = branch1x1->forward(d_input);
float *branch7x7_output = branch7x7_1->forward(d_input); float *branch7x7_output = branch7x7_1->forward(d_input);