Implement avg pooling

2025-11-06 01:34:22 +00:00 · 2024-03-19 22:33:43 +01:00
parent a0fc1b00ae
commit ef63cbd9f1
6 changed files with 92 additions and 45 deletions
--- a/include/layers/max_pooling.cuh
+++ b/include/layers/max_pooling.cuh
@@ -15,7 +15,6 @@ class MaxPooling2D : public SequentialLayer {
        int            nChannels,
        int            poolingSize,
        int            stride,
-        Padding            padding,
        ActivationType activationType
    );
    ~MaxPooling2D();
@@ -27,7 +26,6 @@ class MaxPooling2D : public SequentialLayer {
    int nChannels;
    int poolingSize;
    int stride;
-    int paddingSize;

    int outputSize;
    int gridSize;