Tensorflow模型保存在远程目录中

时间:2018-08-24 11:17:12

标签: tensorflow

我有一个Tensoflow模型,这给我带来了一些麻烦。确实,当我的保存目录位于远程环境(Google Cloud)中时,保存过程就成为了一个真正的负担。它将纪元时间从3s增加到26s。

我正在寻找一种将中间检查点保存在内存中并且仅将最后一个保存到远程目录的方法。

到目前为止,我发现的替代方法是使用内存文件系统,这并不是一个方便的选择,因为我每次运行代码时都必须处理文件系统的创建和安装。

有什么建议吗?

0 个答案:

没有答案