I simply changed the ConvTranspose2d with Conv2d and I get this error. However, the weight initially is assigned manually: https://github.com/nerminsamet/houghnet/blob/master/src/lib/models/networks/houghnet_large_hourglass.py#L276 1