[TF2.0]xxxx.h5模型导入几个报错及解决办法

1.ValueError: Unknown XXX: YYY

解决办法是在tf.keras.models.load_model()括号里面加上
         custom_objects = {“YYY”:XXX对应的函数或者对象}

例如教程里是把网络模型定义成一个class,class名字为MyModel(它是继承了Sequential),这个解决办法是在模型导入的时候加上 custom_objects = {“MyModel”:Sequential},例如以下代码:

new_model=tf.keras.models.load_model("chapter10/models/model.h5",
                                      custom_objects = {"MyModel":Sequential})

有时候还会再报其他的错,长这样的这种类型的错误缺什么加什么一般就管用。
例如报错 ValueError: Unknown activation function:leaky_relu 就加上

new_model=tf.keras.models.load_model("chapter10/models/model.h5",
                                      custom_objects = {"MyModel":Sequential,
                                                        "leaky_relu":tf.nn.leaky_relu})

2.ValueError: Error when checking input:

如果训练模型的时候,第一层没有表明输入张量的维度,导入模型时可能会报错:

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

如果把上述维度添加上再训练,写成 input_shape=(None,784),导入模型时可能会报错:

ValueError: Error when checking input: expected dense_input to have 3 dimensions, but got array with shape (5139, 784)

一种避免出错的做法是训练模型时,用input_dim=784来表明输入数据的维度,例如:

Dense(100,kernel_initializer=he,activation=elu,input_dim=784)
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 5
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 5
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值