weixin_46527744 2020-09-11 10:58 采纳率: 0%
浏览 530

tensorflow2.x 如何用已经保存的模型参数直接预测

1、第一次已经训练好网络并保存了它的参数,第二次不训练网络直接用保存的参数进行预测(前后两次的网络结构一模一样)

2、第一次:

model.save_weights('C:/TF2.1/base_model.h5')

图片说明

3、第二次:

model.load_weights('base_model.h5')

model.predict(x_test)

图片说明

4、报错:You are trying to load a weight file containing 4 layers into a model with 0 layers.

  • 写回答

2条回答 默认 最新

  • threenewbee 2020-09-11 12:45
    关注

    用load_model加载模型
    不要调用 fit
    直接调用 predict

    评论

报告相同问题?

悬赏问题

  • ¥15 想问一下stata17中这段代码哪里有问题呀
  • ¥15 flink cdc无法实时同步mysql数据
  • ¥100 有人会搭建GPT-J-6B框架吗?有偿
  • ¥15 求差集那个函数有问题,有无佬可以解决
  • ¥15 【提问】基于Invest的水源涵养
  • ¥20 微信网友居然可以通过vx号找到我绑的手机号
  • ¥15 寻一个支付宝扫码远程授权登录的软件助手app
  • ¥15 解riccati方程组
  • ¥15 使用rabbitMQ 消息队列作为url源进行多线程爬取时,总有几个url没有处理的问题。
  • ¥15 Ubuntu在安装序列比对软件STAR时出现报错如何解决