I've created a Docker container with all the dependencies listed in your repo. However, when I run the training process I get the following runtime error:
Traceback (most recent call last): File "train.py", line 409, in <module> bat_pc[:, :, :3], bat_pc[:, :, 3:9], global_features, sidx, 0 File "/usr/local/lib/python3.6/dist-packages/torch/nn/modules/module.py", line 550, in __call__ result = self.forward(*input, **kwargs) File "/usr/local/lib/python3.6/dist-packages/torch/nn/parallel/data_parallel.py", line 153, in forward return self.module(*inputs[0], **kwargs[0]) File "/usr/local/lib/python3.6/dist-packages/torch/nn/modules/module.py", line 550, in __call__ result = self.forward(*input, **kwargs) File "/root/gicn_model.py", line 316, in forward grouped_points = query_and_group(xyz.contiguous(), new_xyz.permute(0,2,1).contiguous() , points.permute(0,2,1).contiguous()) File "/usr/local/lib/python3.6/dist-packages/torch/nn/modules/module.py", line 550, in __call__ result = self.forward(*input, **kwargs) File "/root/Pointnet2.PyTorch/pointnet2/pointnet2_utils.py", line 259, in forward new_features = torch.cat([grouped_xyz, grouped_features], dim=1) # (B, C + 3, npoint, nsample) RuntimeError: CUDA error: device-side assert triggered
Hi,
I've created a Docker container with all the dependencies listed in your repo. However, when I run the training process I get the following runtime error:
Traceback (most recent call last): File "train.py", line 409, in <module> bat_pc[:, :, :3], bat_pc[:, :, 3:9], global_features, sidx, 0 File "/usr/local/lib/python3.6/dist-packages/torch/nn/modules/module.py", line 550, in __call__ result = self.forward(*input, **kwargs) File "/usr/local/lib/python3.6/dist-packages/torch/nn/parallel/data_parallel.py", line 153, in forward return self.module(*inputs[0], **kwargs[0]) File "/usr/local/lib/python3.6/dist-packages/torch/nn/modules/module.py", line 550, in __call__ result = self.forward(*input, **kwargs) File "/root/gicn_model.py", line 316, in forward grouped_points = query_and_group(xyz.contiguous(), new_xyz.permute(0,2,1).contiguous() , points.permute(0,2,1).contiguous()) File "/usr/local/lib/python3.6/dist-packages/torch/nn/modules/module.py", line 550, in __call__ result = self.forward(*input, **kwargs) File "/root/Pointnet2.PyTorch/pointnet2/pointnet2_utils.py", line 259, in forward new_features = torch.cat([grouped_xyz, grouped_features], dim=1) # (B, C + 3, npoint, nsample) RuntimeError: CUDA error: device-side assert triggered