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 10:44:08 -04:00
parent bb0a510330
commit 101eabff90
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
run_squad_pytorch.py
View File

@@ -905,8 +905,10 @@ def main():
start_logits, end_logits = model(input_ids, segment_ids, input_mask)
unique_id = [int(eval_features[e.item()].unique_id) for e in example_index]
start_logits = [x.item() for x in start_logits]
end_logits = [x.item() for x in end_logits]
#start_logits = [x.item() for x in start_logits]
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,