fix eval for wt103

a28dfc86 · thomwolf · c03c1268 · a28dfc86
Commit a28dfc86 authored Jan 16, 2019 by thomwolf
Hide whitespace changes
Inline Side-by-side

Showing with 6 additions and 6 deletions

examples/eval_transfo_xl.py examples/eval_transfo_xl.py +6 -6

No files found.
--- 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']:
+    # if args.dataset in ['enwik8', 'text8']:
-        log_str = '| {0} loss {1:5.2f} | {0} bpc {2:9.5f} '.format(
+    #     log_str = '| {0} loss {1:5.2f} | {0} bpc {2:9.5f} '.format(
-            split, loss, loss / math.log(2))
+    #         split, loss, loss / math.log(2))
-    else:
+    # else:
-        log_str = '| {0} loss {1:5.2f} | {0} ppl {2:9.3f} '.format(
+    log_str = '| {0} loss {1:5.2f} | {0} ppl {2:9.3f} '.format(
-            split, loss, math.exp(loss))
+        split, loss, math.exp(loss))
    return log_str
 log_str = ''