Medix-AI/coqui-tts
Medix-AI
/
coqui-tts
mirror of https://github.com/coqui-ai/TTS.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Logger field naming update for layer stats

This commit is contained in:
Eren Golge 2018-12-17 16:38:36 +01:00
parent 22664a52c8
commit b11f307b0e
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
utils/logger.py
View File

@ -12,16 +12,16 @@ class Logger(object):
layer_num = 1 layer_num = 1
for name, param in model.named_parameters(): for name, param in model.named_parameters():
self.writer.add_scalar( self.writer.add_scalar(
"layer{}-ModelParams/{}/max".format(layer_num, name), "layer{}-{}/max".format(layer_num, name),
param.max(), step) param.max(), step)
self.writer.add_scalar( self.writer.add_scalar(
"layer{}-ModelParams/{}/min".format(layer_num, name), "layer{}-{}/min".format(layer_num, name),
param.min(), step) param.min(), step)
self.writer.add_scalar( self.writer.add_scalar(
"layer{}-ModelParams/{}/mean".format(layer_num, name), "layer{}-{}/mean".format(layer_num, name),
param.mean(), step) param.mean(), step)
self.writer.add_scalar( self.writer.add_scalar(
"layer{}-ModelParams/{}/std".format(layer_num, name), "layer{}-{}/std".format(layer_num, name),
param.std(), step) param.std(), step)
self.writer.add_histogram( self.writer.add_histogram(
"layer{}-{}/param".format(layer_num, name), param, step) "layer{}-{}/param".format(layer_num, name), param, step)