feat: layer getting gradient from RNN

modern-fortran · Nov 14, 2023 · 9d2f634 · 9d2f634
1 parent 1bef0aa
commit 9d2f634
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/nf/nf_layer_submodule.f90 b/src/nf/nf_layer_submodule.f90
@@ -379,6 +379,8 @@ pure module function get_gradients(self) result(gradients)
         ! No gradients to get.
       type is (reshape3d_layer)
         ! No gradients to get.
+      type is (rnn_layer)
+        gradients = this_layer % get_gradients()
       class default
         error stop 'get_gradients() with unknown layer type.'
     end select