Python 在循环中运行保存/还原模型函数时出错
在我的程序中,我保存我的模型,然后恢复它 我必须用不同的参数在循环中运行这个程序几次 第一次迭代按预期运行,模型被保存和恢复并正常工作,但在第二次迭代中,模型被训练和保存,但在恢复时,它似乎转到上一个模型 我为每个迭代保留了不同的目录,因此模型的路径是不同的 我的做法如下:Python 在循环中运行保存/还原模型函数时出错,python,tensorflow,Python,Tensorflow,在我的程序中,我保存我的模型,然后恢复它 我必须用不同的参数在循环中运行这个程序几次 第一次迭代按预期运行,模型被保存和恢复并正常工作,但在第二次迭代中,模型被训练和保存,但在恢复时,它似乎转到上一个模型 我为每个迭代保留了不同的目录,因此模型的路径是不同的 我的做法如下: 1。save\u path=self.saver.save(sess,“/tmp/model.ckpt”) 2。saver=tf.train.import\u meta\u graph('/tmp/model.ckpt.me
1。save\u path=self.saver.save(sess,“/tmp/model.ckpt”)
2。saver=tf.train.import\u meta\u graph('/tmp/model.ckpt.meta',clear\u devices=True)
3。saver.restore(sess、tf.train.latest_checkpoint('/tmp/'))
它给出了错误:
InvalidArgumentError:Assign要求两个张量的形状都匹配。左侧形状=[6491024]右侧形状=[6601024]
InvalidArgumentError:从检查点还原失败。这很可能是由于不匹配
当前图形和检查点图形之间的。请确保您没有
根据检查点更改了预期的图形。
lhs形状是在第一次迭代中保存的模型中的一个变量的形状
我不明白他们为什么搞混了,我被卡住了。
非常感谢您的帮助。您使用的是同一个会话还是为每个模型创建一个新会话?@de1会话不同。您使用的是同一个会话还是为每个模型创建一个新会话?@de1会话不同。