tf2.0 ValueError:You are trying to load a weight file containing 2 layers into a model with 0 layers

tensorflow2.0版本,通过继承 tf.keras.Model 类自定义网络结构,当加载之前训练过的模型参数时报错 ValueError:You are trying to load a weight file containing 2 layers into a model with 0 layers。

网上有说改成
model.load_weights('model_name.h5',by_name=True)
试了下还是报错。

后来在github的一些demo里发现,在加载权重前将模型运行一遍将模型初始化就可以了。

model = MyModel()  # 实例化
model(tf.ones(shape=INPUT_SHAPE))  # 随便用个输入跑一下,初始化模型
model.load_weights('model_name.h5')  # 加载权重

最后吐糟一下 tensorflow2.0 坑有点多

评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值