Pytorch数据使用列表的卷积层时报错及解决-RuntimeError: Input type (torch.cuda.HalfTensor) and weight type (torch.FloatTensor) should be same

错误信息

• 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.ModuleList 使用和 list 类似，具体说明见文档

• self.proj = nn.ModuleList([nn.Conv2d(self.in_chans, self.dim[i], kernel_size=self.patch_size,\
                                     stride=self.patch_size) for i in range(len(self.dim))])

• 说实话，当初写的时候，Github copilot 提醒我了，即便它看起来就很像用于 PyTorch 的列表，但我还是没用…