Commit a28dfc86 authored by thomwolf's avatar thomwolf
Browse files

fix eval for wt103

parent c03c1268
......@@ -132,10 +132,10 @@ 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:
# 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
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment