使用 ModuleDict，我有：输入类型（torch.cuda.FloatTensor）

Question

我正在尝试我的__init__功能：        self.downscale_time_conv = np.empty(8, dtype=object)        for i in range(8):            self.downscale_time_conv[i] = torch.nn.ModuleDict({})但在我的forward，我有：        down_out = False        for i in range(8):            if not down_out:                down_out = self.downscale_time_conv[i][side](inputs)            else:                down_out += self.downscale_time_conv[i][side](inputs)我得到：RuntimeError: Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor) should be the samePython火炬张量

Answer 1

      self.downscale_time_conv = torch.nn.ModuleList()

        for i in range(8):

            self.downscale_time_conv.append(torch.nn.ModuleDict({}))

这解决了它。显然我需要使用ModuleList