tpython 3.x - Can't save custom subclassed model - Stack Overflowhttps://stackoverflow.com/questions/51806852/cant-save-custom-subclassed-modeltf2.0目前用户自定义类保存存在问题
解决方案是保存模型的参数 使用save_weights方法,调用的时候使用load_weights方法获取参数,custommodel.load_weights(‘test’)即可,其中custommodel是实例化后的类方法