Remove softmax from last inception layer

2025-11-05 17:34:21 +00:00 · 2024-08-31 23:09:55 +02:00
parent c8557fc0e4
commit 0c8ca44680
1 changed files with 1 additions and 1 deletions
--- 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);
 }