如何在“火车”中间记录图像

Question

您似乎只能通过 train 中的 return 值记录数据。在许多工作流程中，直接在训练函数中间保存图像可能更有意义（例如，保存由生成模型或基于视觉的 MDP 采样的图像）。

有没有简单的方法可以做到这一点？一种想法是尝试找到日志目录并直接写入它，但这会有问题吗？

Answer 1

我猜你问的是在 Trainable:_train() 中使用日志图像。

如果你不在本地模式下，在可训练的范围内，你可以访问 self.logdir 属性来写入图像。这应该会自动同步回您的头节点（如果您是 运行 远程）。