SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

fix eval for wt103

Browse Source
This commit is contained in:
thomwolf
2019-01-16 11:18:19 +01:00
parent c03c12687f
commit a28dfc8659
1 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
examples/eval_transfo_xl.py
View File

@@ -132,12 +132,12 @@ 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
log_str = '' log_str = ''