wassname/TTS
1
0
Fork 0
mirror of https://github.com/wassname/TTS.git synced 2026-09-09 11:16:00 +08:00
Code Issues Packages Projects Releases Wiki Activity

Logger field naming update for layer stats

Browse Source
This commit is contained in:
Eren Golge
2018-12-28 14:22:41 +01:00
parent 806643300c
commit 4abc9ad1bc
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
utils/logger.py
+4 -4
View File
@@ -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)