Skip to content

Commit

Permalink
Fix tensorboard continue_from
Browse files Browse the repository at this point in the history
  • Loading branch information
SiddGururani committed Jun 16, 2017
1 parent cf20b8d commit 386da3c
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion train.py
Original file line number Diff line number Diff line change
Expand Up @@ -185,7 +185,7 @@ def main():
package['loss_results'] is not None and start_epoch > 0: # Previous scores to tensorboard logs
loss_results, cer_results, wer_results = package['loss_results'], package['cer_results'], package[
'wer_results']
for i in range(len(loss_results)):
for i in range(epoch):
info = {
'Avg Train Loss': loss_results[i],
'Avg WER': wer_results[i],
Expand Down

0 comments on commit 386da3c

Please sign in to comment.