diff --git a/modules/deeplearning/icing.py b/modules/deeplearning/icing.py index 3cdc535d9f485b934910f3f565d0ad6d4cef6e87..4c2db1300db70c92c84e0d5a80dbb6a074dc6535 100644 --- a/modules/deeplearning/icing.py +++ b/modules/deeplearning/icing.py @@ -360,6 +360,10 @@ class IcingIntensityNN: fc = build_residual_block(fc, drop_rate, fac*n_hidden, activation, 'Residual_Block_5', doBatchNorm=False) + fc = build_residual_block(fc, drop_rate, fac*n_hidden, activation, 'Residual_Block_4', doBatchNorm=False) + + fc = build_residual_block(fc, drop_rate, fac*n_hidden, activation, 'Residual_Block_5', doBatchNorm=False) + fc = tf.keras.layers.Dense(n_hidden, activation=activation)(fc) fc = tf.keras.layers.BatchNormalization()(fc) print(fc.shape)