Untitled

mail@pastecode.io avatar
unknown
plain_text
3 years ago
1.2 kB
2
Indexable
Never
~/miniconda3/envs/hgcal_dev/lib/python3.8/site-packages/torchsparse/nn/modules/conv.py in forward(self, inputs)
     56 
     57     def forward(self, inputs):
---> 58         return conv3d(inputs,
     59                       self.kernel,
     60                       self.bias,

~/miniconda3/envs/hgcal_dev/lib/python3.8/site-packages/torchsparse/nn/functional/sparseconv.py in conv3d(inputs, kernel, bias, stride, dilation, transpose)
    149             hash_target = sphash(coords)
    150             idx_query = sphashquery(hash_query, hash_target)
--> 151             idx_query = list(convert_neighbor_map_gpu(idx_query))
    152             idx_query[1] = idx_query[1].to('cpu')
    153             sizes = (features.shape[0], new_coords.shape[0])

~/miniconda3/envs/hgcal_dev/lib/python3.8/site-packages/torchsparse/nn/functional/convert_neighbor_map.py in forward(ctx, neighbor_map)
      8     @staticmethod
      9     def forward(ctx, neighbor_map):
---> 10         idx_batch, idx_point = torch.where(neighbor_map != -1)
     11         if 'cuda' in str(neighbor_map.device):
     12             map_converted = torchsparse_cuda.convert_map_forward(

ValueError: not enough values to unpack (expected 2, got 1)