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

4
include/layers/conv2d.cuh
View File

@@ -23,7 +23,7 @@ class Conv2d : public WeightedLayer {
* @param kernelSize Width and height of the convolution kernel
* @param stride Convolution stride
* @param numFilters Number of output filters
* @param padding Padding type ('SAME' or 'VALID')
* @param paddingSize Padding size
* @param activationType Activation function type ('RELU', 'SIGMOID',
* 'SOFTMAX' or 'NONE')
*/
@@ -33,7 +33,7 @@ class Conv2d : public WeightedLayer {
int kernelSize,
int stride,
int numFilters,
Padding padding,
int paddingSize,
ActivationType activationType
);