Mathis/CUDANet
1
0
Fork 0
mirror of https://github.com/lordmathis/CUDANet.git synced 2025-11-05 17:34:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

Unify 2d layer naming

Browse Source
This commit is contained in:
Matúš Námešný
2024-05-20 16:23:58 +02:00
parent 74098b24e3
commit 6dca8ccd3c
12 changed files with 41 additions and 41 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
examples/alexnet/alexnet.cpp
View File

@@ -45,7 +45,7 @@ CUDANet::Model *createModel(
CUDANet::Layers::ActivationType::RELU
);
model->addLayer("features.0", conv1); // Match pytorch naming
CUDANet::Layers::MaxPooling2D *pool1 = new CUDANet::Layers::MaxPooling2D(
CUDANet::Layers::MaxPooling2d *pool1 = new CUDANet::Layers::MaxPooling2d(
56, 64, 3, 2, CUDANet::Layers::ActivationType::NONE
);
model->addLayer("pool1", pool1);
@@ -55,7 +55,7 @@ CUDANet::Model *createModel(
27, 64, 5, 1, 192, 2, CUDANet::Layers::ActivationType::RELU
);
model->addLayer("features.3", conv2);
CUDANet::Layers::MaxPooling2D *pool2 = new CUDANet::Layers::MaxPooling2D(
CUDANet::Layers::MaxPooling2d *pool2 = new CUDANet::Layers::MaxPooling2d(
27, 192, 3, 2, CUDANet::Layers::ActivationType::NONE
);
model->addLayer("pool2", pool2);
@@ -77,7 +77,7 @@ CUDANet::Model *createModel(
13, 256, 3, 1, 256, 1, CUDANet::Layers::ActivationType::RELU
);
model->addLayer("features.10", conv5);
CUDANet::Layers::MaxPooling2D *pool5 = new CUDANet::Layers::MaxPooling2D(
CUDANet::Layers::MaxPooling2d *pool5 = new CUDANet::Layers::MaxPooling2d(
13, 256, 3, 2, CUDANet::Layers::ActivationType::NONE
);
model->addLayer("pool5", pool5);

2
examples/inception_v3/inception_v3.cpp
View File

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