保存包含自定义层的复合模型会导致错误-None类型具有NoneType,但预期其类型是bytes或unicode。
创始人
2024-11-22 05:00:28
0

出现此错误的原因是,无法序列化模型中的自定义层对象。解决方法是定义一个新的类继承自自定义层,并实现get_config()和from_config()方法。get_config()方法返回层的参数配置,from_config()方法使用这些参数重新创建层。将新的类用于替换原始的自定义层即可。示例代码如下:

class CustomLayer(tf.keras.layers.Layer):
    def __init__(self, units, **kwargs):
        self.units = units
        super().__init__(**kwargs)

    def build(self, input_shape):
        self.kernel = self.add_weight(name='kernel',
                                      shape=(input_shape[1], self.units),
                                      initializer='uniform',
                                      trainable=True)

    def call(self, inputs):
        return tf.matmul(inputs, self.kernel)

    def get_config(self):
        config = super().get_config().copy()
        config.update({'units': self.units})
        return config

    @classmethod
    def from_config(cls, config):
        return cls(**config)

# 创建一个包含自定义层的模型,并编译。
inputs = tf.keras.Input(shape=(1,))
x = CustomLayer(64)(inputs)
outputs = tf.keras.layers.Dense(1)(x)
model = tf.keras.Model(inputs, outputs)
model.compile(optimizer='adam',
              loss='mse',
              metrics=['mae'])

# 将新的类用于替换原始的自定义层,并保存模型。
model.layers[1] = CustomLayer.from_config(model.layers[1].get_config())
model.save('model.h5')

相关内容

热门资讯

9分钟了解!福建众娱辅助透视系... 9分钟了解!福建众娱辅助透视系统!总是一直总是有辅助软件(确实有挂)-哔哩哔哩1、福建众娱辅助透视系...
第6分钟了解!新玉海楼游戏茶苑... 第6分钟了解!新玉海楼游戏茶苑!真是一直总是有辅助工具(有挂方略)-哔哩哔哩1、下载好新玉海楼游戏茶...
三分钟了解!微信九游大厅辅助!... 三分钟了解!微信九游大厅辅助!切实是有辅助工具(今日头条)-哔哩哔哩微信九游大厅辅助是不是有人用挂微...
八分钟了解!科乐填大坑辅助器!... 八分钟了解!科乐填大坑辅助器!真是有辅助攻略(有挂透视)-哔哩哔哩1、在科乐填大坑辅助器插件功能辅助...
5分钟了解!蛮籽酉阳大贰辅助!... 5分钟了解!蛮籽酉阳大贰辅助!真是是有辅助攻略(有挂功能)-哔哩哔哩1、玩家可以在蛮籽酉阳大贰辅助线...
三分钟了解!竞技联盟辅助!真是... 您好,竞技联盟辅助这款游戏可以开挂的,确实是有挂的,需要了解加去威信【136704302】很多玩家在...
十分钟了解!辅助软件购买平台!... 十分钟了解!辅助软件购买平台!本来是真的有辅助app(有挂神器)-哔哩哔哩1、完成辅助软件购买平台有...
第七分钟了解!广西友乐app辅... 您好,广西友乐app辅助工具这款游戏可以开挂的,确实是有挂的,需要了解加去威信【136704302】...
第7分钟了解!微乐小程序游戏破... 第7分钟了解!微乐小程序游戏破解器!好像是真的有辅助软件(有挂讲解)-哔哩哔哩1、该软件可以轻松地帮...
第1分钟了解!jj斗地主有透视... 第1分钟了解!jj斗地主有透视吗!切实真的是有辅助攻略(新版有挂)-哔哩哔哩1、在jj斗地主有透视吗...