c-my/transformers
1
0 Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update sequencesummary module

Browse Source
This commit is contained in:
thomwolf committed 2019-07-09 15:46:43 +02:00
1 parent d5481cbe1b
commit d0efbd3cd1
5 files changed
+15 -2
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

pytorch_transformers/tests/modeling_tests_commons.py
+1 -1
View File
@@ -396,7 +396,7 @@ class GPTModelTester(object):
model = self.double_head_model_class(config)
model.eval()
outputs = model(input_ids, mc_token_ids, lm_labels=lm_labels, mc_labels=mc_labels,
token_type_ids=token_type_ids, position_ids=position_ids)
token_type_ids=token_type_ids, position_ids=position_ids)
lm_loss, mc_loss, lm_logits, mc_logits = outputs[:4]
loss = [lm_loss, mc_loss]