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 = ''