Output tensor of ME.SparseTensor has different size with input tensor.

[Dangzheng]

Hi,

Thanks for sharing the code. I'm trying to fine-tune DGR on another dataset. So I modify the dataloader which produces same format data as the data provided by threedmatch_loader. But I found that the output of the ME.SparseTensor has different shape with the input tensor.

To be specific the error lies in this line.

DeepGlobalRegistration/core/trainer.py

Line 633 in 46dd264

sinput0 = ME.SparseTensor(iF0, coords=iC0).to(self.device)

Here is the shape of the tensors, I notice that there are a lot of lines are the same in the tensor iC0. The replicate line issue also happens in the threedmatch data, but the output and the input is same. Do you have any clue about how to solve this issue? Thanks in advance.

Best,

[cjluLeiZhu]

Hi! I am facing a similar issue as well. How did you solve this?

Thanks!
@chrischoy @Dangzheng