From beb59080b37a0826fb2aee2e690d3fa0ea9028d2 Mon Sep 17 00:00:00 2001 From: VictorSanh Date: Fri, 2 Nov 2018 04:36:42 -0400 Subject: [PATCH] Fix size compatibility for model.forward Error was coming from "modeling_pytorch.py", line 484, in forward: start_loss = loss_fct(start_logits, start_positions) --> ValueError: Expected target size (12, 1), got torch.Size([12]) --- run_squad_pytorch.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/run_squad_pytorch.py b/run_squad_pytorch.py index c707a1cd27..b37bee996d 100644 --- a/run_squad_pytorch.py +++ b/run_squad_pytorch.py @@ -840,6 +840,9 @@ def main(): #label_ids = label_ids.to(device) start_positions = start_positions.to(device) end_positions = start_positions.to(device) + + start_positions = start_positions.view(-1, 1) + end_positions = end_positions.view(-1, 1) loss, _ = model(input_ids, segment_ids, input_mask, start_positions, end_positions) loss.backward()