search

Res2Net / Res2Net-PretrainedModels

(ImageNet pretrained models) The official pytorch implemention of the TPAMI paper "Res2Net: A New Multi-scale Backbone Architecture"
https://mmcheng.net/res2net/
1.07k stars 215 forks source link

out = torch.cat((out, self.pool(spx[self.nums])),1) RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 1. Got 16 and 15 in dimension 2 at /pytorch/aten/src/THC/generic/THCTensorMath.cu:87 #19

Closed ghost closed 4 years ago

ghost commented 4 years ago

out = torch.cat((out, self.pool(spx[self.nums])),1) RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 1. Got 16 and 15 in dimension 2 at /pytorch/aten/src/THC/generic/THCTensorMath.cu:87

how to solve it? thanks

gasvn commented 4 years ago

Maybe you should check the input image size.