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

Rework padding size setting

Browse Source
This commit is contained in:
Matúš Námešný
2024-04-20 16:31:28 +02:00
parent dbaab5652e
commit ecf7416f8e
5 changed files with 44 additions and 59 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
src/layers/conv2d.cu
View File

@@ -12,29 +12,17 @@ Conv2d::Conv2d(
int kernelSize,
int stride,
int numFilters,
Padding padding,
int paddingSize,
ActivationType activationType
)
: inputSize(inputSize),
inputChannels(inputChannels),
kernelSize(kernelSize),
stride(stride),
numFilters(numFilters) {
switch (padding) {
case SAME:
outputSize = inputSize;
paddingSize = ((stride - 1) * inputSize - stride + kernelSize) / 2;
break;
numFilters(numFilters),
paddingSize(paddingSize) {
case VALID:
paddingSize = 0;
outputSize = (inputSize - kernelSize) / stride + 1;
break;
default:
break;
}
outputSize = (inputSize - kernelSize + 2 * paddingSize) / stride + 1;
activation = Activation(
activationType, outputSize * outputSize * numFilters