Commit a28dfc86 authored by thomwolf's avatar thomwolf
Browse files

fix eval for wt103

parent c03c1268
...@@ -132,10 +132,10 @@ elif args.split == 'test': ...@@ -132,10 +132,10 @@ elif args.split == 'test':
valid_loss = None valid_loss = None
def format_log(loss, split): 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 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