aRI0U / RandLA-Net-pytorch

PyTorch implementation of RandLA-Net

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Error in class RandLANet

FandosA opened this issue · comments

Hello, in the forward of the RandLANet class, I get an error in this line:
extended_neighbors = neighbors.unsqueeze(-1).expand(-1, x.size(1), -1, 1). When I run the model I get:
RuntimeError: The expanded size of the tensor (1) must match the existing size (312) at non-singleton dimension 3. Target sizes: [-1, 512, -1, 1]. Tensor sizes: [1, 1, 312].

What's happening?

Ok it's solved!