将tf.Saver.save检查点存储在变量（或内存）中

Question

我正在使用Tensorflow并使用tf.Saver将当前“最佳”模型存储在硬盘驱动器上以保持持久性：

saver = tf.train.Saver(max_to_keep=1)

[...]

saver.save(
    sess,
    path_to_file,
    global_step=epoch
)

我的网络相当小，运行速度非常快，GPU上的单个纪元在不到10秒的时间内运行。但是，将模型保存到硬盘驱动器需要一到两分钟，占用了大量时间。

是否可以将模型存储在内存中，以避免占用整个运行时间的这么大一部分？如果我能以某种方式将“最佳”模型存储在内存中一段时间​​，并在我告诉模型时将其转储，我可以通过一个重要因素减少整体运行时间。

我查看了tf.Saver文档和实现，我看不出任何方法可以实现我想要的。是否有其他实现或工具可以做我想做的事情？

Answer 1

我不认为use Mailgun.Client, domain: Application.get_env(:radios, Radios.Mailer)[:mailgun_domain], key: Application.get_env(:radios, Radios.Mailer)[:mailgun_key] 支持这一点。但是，您可以挂载内存中的文件系统（如linux中的tmpfs）并保存到该目录，该目录不应触及任何磁盘。