标签： tensorflow keras
我有一个扩展tf.keras.Model类的模型。在此模型中，我重写了compile和train_step函数，并添加了一些额外的方法。
我正常拟合模型，并使用save_model（）方法保存它。然后，当我使用load_model（）将其加载回时，生成的模型没有保存之前所拥有的额外方法，并且当我调用.compile（）或.train_step（）时，未调用覆盖的方法，但是超类的。
有人知道如何解决此问题吗？