hi
i get below error when i use se res next 50 34x2d model
size mismatch, m1: [80 x 8192], m2: [2048 x 1] at /opt/conda/conda-bld/pytorch_1556653099582/work/aten/src/THC/generic/THCTensorMathBlas.cu:268.
Below is last phase of the model
(2): SEResNeXtBottleneck(
(conv1): Conv2d(2048, 1024, kernel_size=(1, 1), stride=(1, 1), bias=False)
(bn1): BatchNorm2d(1024, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)
(conv2): Conv2d(1024, 1024, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), groups=32, bias=False)
(bn2): BatchNorm2d(1024, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)
(conv3): Conv2d(1024, 2048, kernel_size=(1, 1), stride=(1, 1), bias=False)
(bn3): BatchNorm2d(2048, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)
(relu): ReLU(inplace)
(se_module): SEModule(
(avg_pool): AdaptiveAvgPool2d(output_size=1)
(fc1): Conv2d(2048, 128, kernel_size=(1, 1), stride=(1, 1))
(relu): ReLU(inplace)
(fc2): Conv2d(128, 2048, kernel_size=(1, 1), stride=(1, 1))
(sigmoid): Sigmoid()
)
)
)
(avg_pool): AvgPool2d(kernel_size=7, stride=1, padding=0)
(last_linear): Sequential(
(0): view() #x.view(x.size(0),-1)
(1): Linear(in_features=2048, out_features=1, bias=True)
)