bearpaw/pytorch-pose

`RuntimeError` in `HourglassNet` because of possibly incorrect `scale_factor` for `F.interpolate`

joshuacwnewton opened this issue · 2 comments

joshuacwnewton commented

Error

I am trying to export HourglassNet to .onnx using torch.onnx.export. I am getting the following error:

 File "", line, in _hour_glass_forward
   out = up1 + up2
 RuntimeError: The size of tensor a (3) must match the size of tensor b (2) at non-singleton dimension 3
joshuacwnewton commented

Please disregard this issue. It was user error.

bearpaw commented

Thanks for looking into this and welcome to push this again once you found a solution.

  • 👍1