Python 报错记录——ValueError: Unknown layer:XXXXX
利用keras导入自定义模型时会报错:
ValueError: Unknown layer:Mandist
比如我的源代码是:
model = tf.keras.models.load_model('en_SiameseLSTM.h5')
由于原先训练的模型有自定义层Mandist,keras导入时识别不出来就会报错。因此,需要加入
model = tf.keras.models.load_model(