Allocate activation on heap

2025-11-07 18:24:26 +00:00 · 2024-04-22 18:59:16 +02:00
parent 26cea9b12c
commit a32c737785
10 changed files with 17 additions and 15 deletions
--- a/include/layers/conv2d.cuh
+++ b/include/layers/conv2d.cuh
@@ -120,8 +120,7 @@ class Conv2d : public WeightedLayer {
    float* d_weights;
    float* d_biases;

-    // Kernels
-    Activation activation;
+    Activation* activation;

    /**
     * @brief Initialize weights of the convolutional layer with zeros