Traceback (most recent call last):
File "/data/yechangxin/code/PAConv/scene_seg/tool/train.py", line 326, in
main()
File "/data/yechangxin/code/PAConv/scene_seg/tool/train.py", line 146, in main
loss_train, mIoU_train, mAcc_train, allAcc_train = train(train_loader, model, criterion, optimizer, epoch, args.get('correlation_loss', False))
File "/data/yechangxin/code/PAConv/scene_seg/tool/train.py", line 201, in train
output = model(input)
File "/home/yechangxin/anaconda3/envs/torch2/lib/python3.7/site-packages/torch/nn/modules/module.py", line 1051, in _call_impl
return forward_call(*input, *kwargs)
File "/data/yechangxin/code/PAConv/scene_seg/model/pointnet2/pointnet2_paconv_seg.py", line 74, in forward
li_xyz, li_features = self.SA_modules[i](l_xyz[i], l_features[i])
File "/home/yechangxin/anaconda3/envs/torch2/lib/python3.7/site-packages/torch/nn/modules/module.py", line 1051, in _call_impl
return forward_call(input, **kwargs)
File "/data/yechangxin/code/PAConv/scene_seg/model/pointnet2/pointnet2_paconv_modules.py", line 160, in forward
new_xyz_idx = pointops.furthestsampling(xyz, self.npoint) # (B, N1)
File "/data/yechangxin/code/PAConv/scene_seg/lib/pointops/functions/pointops.py", line 55, in forward
pointops_cuda.furthestsampling_cuda(b, n, m, xyz, temp, idx)
TypeError: furthestsampling_cuda(): incompatible function arguments. The following argument types are supported:
Traceback (most recent call last): File "/data/yechangxin/code/PAConv/scene_seg/tool/train.py", line 326, in
main()
File "/data/yechangxin/code/PAConv/scene_seg/tool/train.py", line 146, in main
loss_train, mIoU_train, mAcc_train, allAcc_train = train(train_loader, model, criterion, optimizer, epoch, args.get('correlation_loss', False))
File "/data/yechangxin/code/PAConv/scene_seg/tool/train.py", line 201, in train
output = model(input)
File "/home/yechangxin/anaconda3/envs/torch2/lib/python3.7/site-packages/torch/nn/modules/module.py", line 1051, in _call_impl
return forward_call(*input, *kwargs)
File "/data/yechangxin/code/PAConv/scene_seg/model/pointnet2/pointnet2_paconv_seg.py", line 74, in forward
li_xyz, li_features = self.SA_modules[i](l_xyz[i], l_features[i])
File "/home/yechangxin/anaconda3/envs/torch2/lib/python3.7/site-packages/torch/nn/modules/module.py", line 1051, in _call_impl
return forward_call(input, **kwargs)
File "/data/yechangxin/code/PAConv/scene_seg/model/pointnet2/pointnet2_paconv_modules.py", line 160, in forward
new_xyz_idx = pointops.furthestsampling(xyz, self.npoint) # (B, N1)
File "/data/yechangxin/code/PAConv/scene_seg/lib/pointops/functions/pointops.py", line 55, in forward
pointops_cuda.furthestsampling_cuda(b, n, m, xyz, temp, idx)
TypeError: furthestsampling_cuda(): incompatible function arguments. The following argument types are supported:
Invoked with: 4, 4096, tensor(1024), tensor([[[ 0.1788, -0.0732, 2.0553], [ 0.4895, 0.0057, 1.2836], [ 0.4620, 0.1135, 0.3648], ..., [-0.4602, 0.2132, 0.0284], [-0.1321, -0.0043, 0.0203], [-0.0581, 0.2376, 0.0274]],