Efficientnet-V2 export to ONNX fails

[Mypathissional]

Hi,

I have been trying to export the v2 model to onnx and it fails. Do you know how to fix this?

tf.keras.backend.clear_session()
model = effnetv2_model.EffNetV2Model(model_name=MODEL)
_ = model(tf.ones([1, 224, 224, 3]), training=False)
model.load_weights(ckpt_path)
cfg = model.cfg
onnx_model = keras2onnx.convert_keras(model, model.name)