Merge pull request #418 from jybaek/fixed-best_wer

Fixed the best_wer problem in continue_from option.
SeanNaren · May 20, 2019 · aa59401 · aa59401
2 parents d14273a + 84331d6
commit aa59401
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/train.py b/train.py
@@ -164,6 +164,7 @@ def update(self, val, n=1):
  avg_loss = int(package.get('avg_loss', 0))
  loss_results, cer_results, wer_results = package['loss_results'], package['cer_results'], \
  package['wer_results']
+ best_wer = wer_results[start_epoch]
  if main_proc and args.visdom: # Add previous scores to visdom graph
  visdom_logger.load_previous_values(start_epoch, package)
  if main_proc and args.tensorboard: # Previous scores to tensorboard logs