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

Refactor Backend and Layer interfaces

Browse Source
This commit is contained in:
Matúš Námešný
2025-11-18 18:27:57 +01:00
parent 25670f90c4
commit 6340b27055
23 changed files with 154 additions and 201 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
examples/inception_v3/inception_modules.cpp
View File

@@ -213,7 +213,7 @@ InceptionB::InceptionB(
branch3x3->getOutputSize(), branch3x3dbl_3->getOutputSize()
);
concat_2 = new CUDANet::Layers::Concat(
concat_1->getOutputSize(), branchPool->getOutputSize()
concat_1->getOutputSize(), branchPool->get_output_size()
);
outputSize = concat_2->getOutputSize();
@@ -441,7 +441,7 @@ InceptionD::InceptionD(
branch3x3_2->getOutputSize(), branch7x7x3_4->getOutputSize()
);
concat_2 = new CUDANet::Layers::Concat(
concat_1->getOutputSize(), branchPool->getOutputSize()
concat_1->getOutputSize(), branchPool->get_output_size()
);
outputSize = concat_2->getOutputSize();
@@ -707,7 +707,7 @@ InceptionV3::InceptionV3(
addLayer("AveragePool", avgpool);
fc = new CUDANet::Layers::Dense(
avgpool->getOutputSize(), 1000, CUDANet::Layers::ActivationType::NONE
avgpool->get_output_size(), 1000, CUDANet::Layers::ActivationType::NONE
);
addLayer("fc", fc);
}