the error of "call() got an unexpected keyword argument 'training'",when convert to tflite

[xieshenru]

when convert to tflite,the error of "call() got an unexpected keyword argument 'training'" appear,how to resolve it?

import tensorflow as tf import keras from keras.layers import BatchNormalization, ReLU from keras.backend import pool2d from utils import focal_loss

modelpath = r'./model/c3ae_model_v2.h5' model = keras.models.load_model(modelpath, custom_objects={"pool2d": pool2d, "ReLU": ReLU, "BatchNormalization": BatchNormalization, "tf": tf, "focal_loss_fixed": focal_loss([1] * 12)}) converter = tf.lite.TFLiteConverter.from_keras_model(model)

tflite_model = converter.convert() savepath = r'./model.tflite' open(savepath, "wb").write(tflite_model) the above is the code I used,how to resolve this error?

[StevenBanama]

keras not test yet! you should use tf. keras, which tf=2.1