Как добавить сводную оп в тензорную доску в rllib? - PullRequest
Новая
       12

Как добавить сводную оп в тензорную доску в rllib?

0 голосов
/ 19 апреля 2019

Я новичок в 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. Как я могу добавить эту аннотацию для выполнения во время тренинга?

...