如何在rllib中将摘要操作添加到张量板？

Question

我是rllib的新手，并且正在使用自定义模型和DQNTrainer。我想将输入图像的摘要放入Tensorboard日志中。

我在创建模型时试图定义摘要操作，希望tf.summary.merge_all（）将添加摘要操作。

class CustomModel(Model):
    def _build_layers_v2(self, input_dict, num_outputs, options):
        self.obs_in = input_dict["obs"]
        self.pre_actions = self.obs_in[:, -10:]
        self.observations = tf.reshape(self.obs_in[:, :-10], (-1, 84, 84, 3))
        sumamry_op = tf.summary.image('input_image', self.observations)

这与如何将图像摘要添加到张量板无关。我的目标是向rllib框架添加新的摘要 如何添加要在培训期间执行的摘要操作？