张量流改变检查点位置

时间:2016-07-13 05:10:18

标签: python machine-learning tensorflow

我想知道是否有一种方式来指示您的训练模型(在张量流中)或一般存储检查点文件的张量流配置,我正在转换神经网络并得到错误: InternalError:写入错误(tmp)检查点文件:/tmp/tmpn2cWXm/model.ckpt-500-00000-of-00001.tempstate12392765014661958578:资源耗尽 和 错误:tensorflow:在tf.learn最终检查点期间出现异常。

而且我正在获取有关磁盘空间不足的操作系统警报(debian linux)所以我认为问题是我的磁盘已满了检查点文件但是我的磁盘中有多个分区,有足够的空间并且想要移动检查点文件那里。

谢谢!

2 个答案:

答案 0 :(得分:0)

您可以将保存路径指定为tf.train.Saver.save(sess, 'your/save/path', ...)的第二个参数。同样,您可以将恢复路径的先前保存的变量恢复为tf.train.Saver.restore(sess, 'your/restore/path')的第二个参数。

有关详细信息,请参阅TensorFlow Saver documentationthese saving and restoring examples

答案 1 :(得分:0)

这是一个非常老的查询,我今天在寻找其他问题时偶然发现。无论如何,我想我会放下自己的想法,以防将来对某人有所帮助。

您可以在构造回归器时指定模型目录。这可以是文件系统上您具有写许可权和足够空间的任何位置(事件文件需要很多空间)。例如:

$username = $username.$unique_id; //Append username and unique_id