search

bearpaw / pytorch-pose

A PyTorch toolkit for 2D Human Pose Estimation.
GNU General Public License v3.0
1.11k stars 253 forks source link

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

Closed joshuacwnewton closed 3 years ago

joshuacwnewton commented 3 years ago

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 3 years ago

Please disregard this issue. It was user error.

bearpaw commented 3 years ago

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