如何在火车中间记录图像

Question

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

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

Answer 1

我猜您是在询问有关在Trainable:_train()中使用日志记录图像的问题。

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