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

Rename batchnorm

Browse Source
This commit is contained in:
Matúš Námešný
2024-05-20 13:05:48 +02:00
parent 94df063dcd
commit 6f8b5f4081
4 changed files with 20 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/inception_v3/inception_v3.cpp
View File

@@ -30,7 +30,7 @@ class BasicConv2d : public CUDANet::Module {
int batchNormSize = conv->getOutputSize();
CUDANet::Layers::BatchNorm *batchNorm = new CUDANet::Layers::BatchNorm(
CUDANet::Layers::BatchNorm2D *batchNorm = new CUDANet::Layers::BatchNorm2D(
batchNormSize, outputChannels, 1e-3f,
CUDANet::Layers::ActivationType::RELU
);