Show test BLEU
@@ -174,7 +174,7 @@ if __name__ == "__main__":
writer.add_scalar("bleu_sample", np.mean(bleus_sample), batch_idx)
writer.add_scalar("skipped_samples", skipped_samples / total_samples, batch_idx)
writer.add_scalar("epoch", batch_idx, epoch)
- log.info("Epoch %d", epoch)
+ log.info("Epoch %d, test BLEU: %.3f", epoch, bleu_test)
if best_bleu is None or best_bleu < bleu_test:
best_bleu = bleu_test
log.info("Best bleu updated: %.4f", bleu_test)
