Hi all,
I think I managed to track down the problem to this line:
scaled_input = tf.keras.layers.multiply([input_layer, scale])

TF2 doesn’t like this multiplication. Is it because scale is not a layer?
I guess to fix this I could make a new layer or use a Lambda layer that does the multiplication between the inputs and a scalar.
Thanks!