diff --git a/mt_with_external_memory/train.py b/mt_with_external_memory/train.py index 6dd9187150799507f2160bf9d4ed7a52114fcd42..ecb6d18331f21a7eeec07c78fa856149e2056bfb 100644 --- a/mt_with_external_memory/train.py +++ b/mt_with_external_memory/train.py @@ -135,7 +135,7 @@ def train(): sys.stdout.flush() if isinstance(event, paddle.event.EndPass): result = trainer.test(reader=test_batch_reader, feeding=feeding) - print "Pass: %d, TestCost: %f, %s" % (event.pass_id, event.cost, + print "Pass: %d, TestCost: %f, %s" % (event.pass_id, result.cost, result.metrics) with gzip.open("checkpoints/params.pass-%d.tar.gz" % event.pass_id, 'w') as f: