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

Fix size compatibility for model.forward

Browse Source 
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])
This commit is contained in:
VictorSanh
2018-11-02 04:36:42 -04:00
parent 8cbe7d6af8
commit beb59080b3
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
run_squad_pytorch.py
View File

@@ -841,6 +841,9 @@ def main():
start_positions = start_positions.to(device) start_positions = start_positions.to(device)
end_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, _ = model(input_ids, segment_ids, input_mask, start_positions, end_positions)
loss.backward() loss.backward()
optimizer.step() optimizer.step()