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

Fix bug writing predictions in run_squad_pytorch

Browse Source
This commit is contained in:
VictorSanh
2018-11-02 11:29:45 -04:00
parent ee29871f8d
commit 38f740a1d5
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
run_squad_pytorch.py
View File

@@ -910,8 +910,8 @@ def main():
#end_logits = [x.item() for x in end_logits] #end_logits = [x.item() for x in end_logits]
end_logits = [x.view(-1).detach().cpu().numpy() for x in end_logits] end_logits = [x.view(-1).detach().cpu().numpy() for x in end_logits]
for idx, i in enumerate(unique_id): for idx, i in enumerate(unique_id):
s = start_logits[idx] s = [float(x) for x in start_logits[idx]]
e = end_logits[idx] e = [float(x) for x in end_logits[idx]]
all_results.append( all_results.append(
RawResult( RawResult(
unique_id=i, unique_id=i,