We are using TF.2.4.0 and have multiple models generated as follows:
model = models [model_name] model.compile (optimizer = tf.keras.optimizers.RMSprop (lr=0.