From b11f307b0e4f3a4e77fd897d420fca0341120376 Mon Sep 17 00:00:00 2001 From: Eren Golge Date: Mon, 17 Dec 2018 16:38:36 +0100 Subject: [PATCH] Logger field naming update for layer stats --- utils/logger.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/utils/logger.py b/utils/logger.py index c8cfcf28..2b1e262d 100644 --- a/utils/logger.py +++ b/utils/logger.py @@ -12,16 +12,16 @@ class Logger(object): layer_num = 1 for name, param in model.named_parameters(): self.writer.add_scalar( - "layer{}-ModelParams/{}/max".format(layer_num, name), + "layer{}-{}/max".format(layer_num, name), param.max(), step) self.writer.add_scalar( - "layer{}-ModelParams/{}/min".format(layer_num, name), + "layer{}-{}/min".format(layer_num, name), param.min(), step) self.writer.add_scalar( - "layer{}-ModelParams/{}/mean".format(layer_num, name), + "layer{}-{}/mean".format(layer_num, name), param.mean(), step) self.writer.add_scalar( - "layer{}-ModelParams/{}/std".format(layer_num, name), + "layer{}-{}/std".format(layer_num, name), param.std(), step) self.writer.add_histogram( "layer{}-{}/param".format(layer_num, name), param, step)