From 0c8ca44680673632ea4a737d25daf36ca7fa688e Mon Sep 17 00:00:00 2001 From: LordMathis Date: Sat, 31 Aug 2024 23:09:55 +0200 Subject: [PATCH] Remove softmax from last inception layer --- examples/inception_v3/inception_modules.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/inception_v3/inception_modules.cpp b/examples/inception_v3/inception_modules.cpp index 3cc639f..8bdf203 100644 --- a/examples/inception_v3/inception_modules.cpp +++ b/examples/inception_v3/inception_modules.cpp @@ -707,7 +707,7 @@ InceptionV3::InceptionV3( addLayer("AveragePool", avgpool); fc = new CUDANet::Layers::Dense( - avgpool->getOutputSize(), 1000, CUDANet::Layers::ActivationType::SOFTMAX + avgpool->getOutputSize(), 1000, CUDANet::Layers::ActivationType::NONE ); addLayer("fc", fc); }