I want to train the model and batch_size = 1, so I add drop_last = True to my training dataloader. However, the error is still there.
And the error doesn’t happen in the backbone. It existed at deeplabV3-assp-x5 = F.interpolate(self.avg_pool(x), size=x.size()[2:], mode='bilinear', align_corners=True)