RuntimeError: Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor) should be the same
定位报错位置:
conv3 = nn.Conv2d(96, 24, 3, 1, 1, bias=False)
修改为
self.conv3 = nn.Conv2d(96, 24, 3, 1, 1, bias=False)
RuntimeError: Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor) should be the
最新推荐文章于 2024-05-31 16:10:34 发布