Bug fix

Author: Riccardo231

src/nf/nf_conv1d_layer_submodule.f90

@@ -157,8 +157,8 @@ module function get_params(self) result(params)
     class(conv1d_layer), intent(in), target :: self
     real, allocatable :: params(:)
     real, pointer :: w_(:) => null()
-    w_(1:size(self % z)) => self % z
-    params = [ w_]
+    w_(1:size(self % kernel)) => self % kernel
+    params = [ w_, self % biases]
   end function get_params
 
   module function get_gradients(self) result(gradients)

src/nf/nf_conv2d_layer_submodule.f90

@@ -195,11 +195,11 @@ module function get_params(self) result(params)
 
     real, pointer :: w_(:) => null()
 
-    w_(1:size(self % z)) => self % z
+    w_(1:size(self % kernel)) => self % kernel
 
     params = [ &
-      w_ &
-      !self % biases &
+      w_, &
+      self % biases &
     ]
 
   end function get_params

test/test_conv2d_network.f90

@@ -60,7 +60,7 @@ program test_conv2d_network
       call cnn % forward(sample_input)
       call cnn % backward(y)
       call cnn % update(optimizer=sgd(learning_rate=1.))
-      o = cnn % layers(2) % get_params()
+      o = cnn % layers(3) % get_params()
       print *, o
       if (all(abs(cnn % predict(sample_input) - y) < tolerance)) exit
     end do