加载TensorFlow冻结的模型(作为* .pb文件),并继续对其进行训练

时间:2019-04-16 06:33:16

标签: python tensorflow deep-learning reinforcement-learning openai-gym

我正在Uber的Atari-model-zoo存储库(https://github.com/uber-research/atari-model-zoo)上做一些工作。

基本上,我正在尝试从训练的早期阶段加载冻结的模型,并继续在本地进行训练,同时在图形上执行一些操作(偶尔进行前馈等等)。

Uber在训练的不同阶段将冻结的模型作为* .pb文件提供,但我不了解如何重新创建图形并将权重和其他元数据参数加载到其中。

理想情况下,我想做类似的事情:

  1. m = import_model('。pb')

  2. 用于纪元

    2.1批处理:

    2.1.1 sess.run(m.training_op, feed_dict={x,y})
    

谢谢。

0 个答案:

没有答案