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
评论列表
文章目录