more console printing formatting

This commit is contained in:
erogol
2020-04-29 11:58:51 +02:00
parent 091711459d
commit 67420eeb86
2 changed files with 4 additions and 3 deletions
+1 -1
View File
@@ -586,7 +586,7 @@ def main(args): # pylint: disable=redefined-outer-name
model.decoder.set_r(r)
if c.bidirectional_decoder:
model.decoder_backward.set_r(r)
print("\n > Number of output frames:", model.decoder.r)
print("\n > Number of output frames:", model.decoder.r)
train_avg_loss_dict, global_step = train(model, criterion, optimizer,
optimizer_st, scheduler, ap,