nec_agent.py 文件源码

python
阅读 26 收藏 0 点赞 0 评论 0

项目:nec_tensorflow 作者: toth-adam 项目源码 文件源码
def _tensorboard_summary_writer(self, batch_total_loss):
        # if self.global_step == self.optimization_start:
        if self.create_list_for_total_losses:
            self.create_list_for_total_losses = False
            self._loss_list = []
            self._mean_size = 0

        self._loss_list.append(batch_total_loss)
        self._mean_size += 1
        if self._mean_size % 10 == 0:
            mean_total_loss = np.mean(self._loss_list)
            summary = tf.Summary()
            summary.value.add(tag='Total Loss', simple_value=float(mean_total_loss))
            self.summary_writer.add_summary(summary, self.global_step)
            self.summary_writer.flush()
            self._loss_list = []
            self._mean_size = 0
评论列表
文章目录


问题


面经


文章

微信
公众号

扫码关注公众号