feat: layer setting params for RNN

modern-fortran · Nov 14, 2023 · 3fa5281 · 3fa5281
1 parent 9d2f634
commit 3fa5281
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/nf/nf_layer_submodule.f90 b/src/nf/nf_layer_submodule.f90
@@ -438,6 +438,9 @@ module subroutine set_params(self, params)
         write(stderr, '(a)') 'Warning: calling set_params() ' &
           // 'on a zero-parameter layer; nothing to do.'
 
+      type is (rnn_layer)
+        call this_layer % set_params(params)
+
           class default
         error stop 'set_params() with unknown layer type.'
     end select