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-07-26 19:33:38 发布
本文探讨了在使用PyTorch构建深度学习模型时遇到的RuntimeError,焦点在于`torch.cuda.FloatTensor`和`torch.FloatTensor`类型的不匹配。通过实例说明如何修正`nn.Conv2d`层的bias参数,以确保正确输入类型。
摘要由CSDN通过智能技术生成