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

Debug run_squad_pytorch

Browse Source
This commit is contained in:
VictorSanh
2018-11-02 11:07:32 -04:00
parent 101eabff90
commit ee29871f8d
1 changed files with 15 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
run_squad_pytorch.py
View File

@@ -909,11 +909,21 @@ def main():
start_logits = [x.view(-1).detach().cpu().numpy() for x in start_logits]
#end_logits = [x.item() for x in end_logits]
end_logits = [x.view(-1).detach().cpu().numpy() for x in end_logits]
all_results.append(
RawResult(
unique_id=unique_id,
start_logits=start_logits,
end_logits=end_logits))
for idx, i in enumerate(unique_id):
s = start_logits[idx]
e = end_logits[idx]
all_results.append(
RawResult(
unique_id=i,
start_logits=s,
end_logits=e
)
)
# all_results.append(
# RawResult(
# unique_id=unique_id,
# start_logits=start_logits,
# end_logits=end_logits))
output_prediction_file = os.path.join(args.output_dir, "predictions.json")
output_nbest_file = os.path.join(args.output_dir, "nbest_predictions.json")