最新文章
Pytorch数据使用列表的卷积层时报错及解决-RuntimeError: Input type (torch.cuda.HalfTensor) and weight type (torch.FloatTensor) should be same
已阅: 205 错误信息RuntimeError: Input type (torch.cuda.HalfTensor) and weight type (torch.FloatTensor) should be same场景描述数据调用 list 里的卷积层时出现数据不匹配错误。self.proj = [nn.Conv2d(self.in_chans, self.dim[i], kernel_size=self.patch_size,\ stride=self.patch_size) for i in range(len(self.dim))]解决方法将 list 改为 nn.ModuleList,nn.Mo……
(◕ܫ◕)
评论于 MMDet实例分割loss_rpn_bbox为nan但其它loss正常的解决