From a28dfc86599b975df845a8ca9331c87af301c6b1 Mon Sep 17 00:00:00 2001 From: thomwolf Date: Wed, 16 Jan 2019 11:18:19 +0100 Subject: [PATCH] fix eval for wt103 --- examples/eval_transfo_xl.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/examples/eval_transfo_xl.py b/examples/eval_transfo_xl.py index 92979d1e4a..e67efd3a68 100644 --- a/examples/eval_transfo_xl.py +++ b/examples/eval_transfo_xl.py @@ -132,12 +132,12 @@ elif args.split == 'test': valid_loss = None def format_log(loss, split): - if args.dataset in ['enwik8', 'text8']: - log_str = '| {0} loss {1:5.2f} | {0} bpc {2:9.5f} '.format( - split, loss, loss / math.log(2)) - else: - log_str = '| {0} loss {1:5.2f} | {0} ppl {2:9.3f} '.format( - split, loss, math.exp(loss)) + # if args.dataset in ['enwik8', 'text8']: + # log_str = '| {0} loss {1:5.2f} | {0} bpc {2:9.5f} '.format( + # split, loss, loss / math.log(2)) + # else: + log_str = '| {0} loss {1:5.2f} | {0} ppl {2:9.3f} '.format( + split, loss, math.exp(loss)) return log_str log_str = ''