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

Remove softmax from last inception layer

Browse Source
This commit is contained in:
Matúš Námešný
2024-08-31 23:09:55 +02:00
parent c8557fc0e4
commit 0c8ca44680
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/inception_v3/inception_modules.cpp
View File

@@ -707,7 +707,7 @@ InceptionV3::InceptionV3(
addLayer("AveragePool", avgpool); addLayer("AveragePool", avgpool);
fc = new CUDANet::Layers::Dense( fc = new CUDANet::Layers::Dense(
avgpool->getOutputSize(), 1000, CUDANet::Layers::ActivationType::SOFTMAX avgpool->getOutputSize(), 1000, CUDANet::Layers::ActivationType::NONE
); );
addLayer("fc", fc); addLayer("fc", fc);
} }