diff --git a/src/nf/nf_layer_submodule.f90 b/src/nf/nf_layer_submodule.f90 index ca18f9c7..3f281411 100644 --- a/src/nf/nf_layer_submodule.f90 +++ b/src/nf/nf_layer_submodule.f90 @@ -359,9 +359,14 @@ impure elemental module subroutine print_info(self) if (.not. self % name == 'input') & print '("Input shape: ", *(i0, 1x))', self % input_layer_shape print '("Output shape: ", *(i0, 1x))', self % layer_shape - print '("Parameters: ", i0)', self % get_num_params() - if (.not. self % name == 'input') & + if (.not. self % name == 'dropout') & + print '("Parameters: ", i0)', self % get_num_params() + if (.not. (self % name == 'input' .or. self % name == 'dropout')) & print '("Activation: ", a)', self % activation + select type (this_layer => self % p) + type is (dropout_layer) + print '("Dropout rate: ", f0.2)', this_layer % dropout_rate + end select print * end subroutine print_info