Mathis/CUDANet
1
0
Fork 0
mirror of https://github.com/lordmathis/CUDANet.git synced 2025-11-06 09:44:28 +00:00
Code Issues Packages Projects Releases Wiki Activity

Implement Inception block D

Browse Source
This commit is contained in:
Matúš Námešný
2024-05-27 21:30:42 +02:00
parent df47a31f36
commit 3955ddd888
2 changed files with 159 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
include/layers/max_pooling.cuh
View File

@@ -9,11 +9,11 @@ namespace CUDANet::Layers {
class MaxPooling2d : public SequentialLayer, public TwoDLayer {
public:
MaxPooling2d(
shape2d inputSize,
shape2d inputSize,
int nChannels,
shape2d poolingSize,
shape2d stride,
shape2d padding,
shape2d poolingSize,
shape2d stride,
shape2d padding,
ActivationType activationType
);
~MaxPooling2d();
@@ -38,7 +38,7 @@ class MaxPooling2d : public SequentialLayer, public TwoDLayer {
private:
shape2d inputSize;
int nChannels;
int nChannels;
shape2d poolingSize;
shape2d stride;
shape2d padding;