TernausNetV2 RuntimeError: Given groups=1, weight of size [64, 11, 3, 3], expected input[1, 6 72, 672, 11] to have 11 channels, but got 672 channels instead

RuntimeError: Given groups=1, weight of size [64, 11, 3, 3], expected input[1, 6 72, 672, 11] to have 11 channels, but got 672 channels instead

Open tonyboston-au opened this issue 6 years ago • 0 comments

Running Demo.ipynb yields runtime error at line: prediction = torch.sigmoid(model(input_img)).data[0].cpu().numpy()

RuntimeError: Given groups=1, weight of size [64, 11, 3, 3], expected input[1, 6 72, 672, 11] to have 11 channels, but got 672 channels instead

Running: python3/3.6.7 pytorch/0.4.1-py36 cuda/9.0 gcc/4.9.0 opencv-python/3.4.4 numpy/1.15.4

Any suggestions appreciated.

Dec 13 '18 05:12 tonyboston-au