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

Question

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

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

Answer 1

我猜你是在问关于在Trainable:_train().

如果您不在本地模式下，则在可训练对象中，您可以访问self.logdir要写入图像的属性。这应该会自动同步回您的头节点（如果您正在远程运行）。