Debug run_squad_pytorch

ee29871f · VictorSanh · 101eabff · ee29871f
Commit ee29871f authored Nov 02, 2018 by VictorSanh
Hide whitespace changes
Inline Side-by-side

Showing with 15 additions and 5 deletions

run_squad_pytorch.py run_squad_pytorch.py +15 -5

No files found.
--- a/run_squad_pytorch.py
+++ b/run_squad_pytorch.py
@@ -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(
+            for idx, i in enumerate(unique_id):
-                RawResult(
+                s = start_logits[idx]
-                    unique_id=unique_id,
+                e = end_logits[idx]
-                    start_logits=start_logits,
+                all_results.append(
-                    end_logits=end_logits))
+                    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")