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); }