Skip to content

GitLab

Commit 101eabff authored by VictorSanh's avatar VictorSanh
Browse files

Debug run_squad_pytorch

parent bb0a5103
Show whitespace changes
Inline Side-by-side
Showing with 4 additions and 2 deletions
run_squad_pytorch.py
View file @ 101eabff
...@@ -905,8 +905,10 @@ def main(): ...@@ -905,8 +905,10 @@ def main():
start_logits, end_logits = model(input_ids, segment_ids, input_mask) 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] unique_id = [int(eval_features[e.item()].unique_id) for e in example_index]
start_logits = [x.item() for x in start_logits] #start_logits = [x.item() for x in start_logits]
end_logits = [x.item() for x in end_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( all_results.append(
RawResult( RawResult(
unique_id=unique_id, unique_id=unique_id,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment