Sizes of tensors must match except in dimension 1. Expected size 44 but got size 45 for tensor numbe
项目场景:跑U-net网络的时候,有一步是torch.cat()操作,出现下面是代码class UNet(nn.Module): def __init__(self, in_channels, out_channels): super(UNet, self).__init__() self.conv1 = DoubleConv(in_channels, 32) self.pool1 = nn.MaxPool2d(2) self.
原创
2022-05-22 15:18:03 ·
14043 阅读 ·
8 评论