Skip to content
Snippets Groups Projects

Resolve "release v2.3.0"

Merged Ghost User requested to merge release_v2.3.0 into master
1 file
+ 6
0
Compare changes
  • Side-by-side
  • Inline
@@ -74,6 +74,12 @@ class BranchedInputResNet(BranchedInputCNN):
kernel_initializer = self._initializer.get(activation_type, "glorot_uniform")
layer_opts["kernel_initializer"] = kernel_initializer
follow_up_layer = activation
if self.bn is True and layer_type.lower() != "residual_block":
another_layer = keras.layers.BatchNormalization
if activation_type in ["relu", "linear", "prelu", "leakyrelu"]:
follow_up_layer = (another_layer, follow_up_layer)
else:
follow_up_layer = (follow_up_layer, another_layer)
regularizer_type = layer_opts.pop("kernel_regularizer", None)
if regularizer_type is not None:
layer_opts["kernel_regularizer"] = self._set_regularizer(regularizer_type, **self.kwargs)
Loading