Traceback (most recent call last):
  File "train.py", line 75, in <module>
    train(args)
  File "train.py", line 64, in train
    output =  CIA_interface.cia_forward(batch ,epoch ,i)
  File "/notebooks/E2E/cia_interface.py", line 90, in cia_forward
    losses =  self.model(example, return_loss = True)
  File "/opt/conda/envs/btngan1/lib/python3.8/site-packages/torch/nn/modules/module.py", line 722, in _call_impl
    result = self.forward(*input, **kwargs)
  File "/notebooks/cia/det3d/models/detectors/voxelnet.py", line 38, in forward
    return self.bbox_head.loss(example, preds)
  File "/notebooks/cia/det3d/models/bbox_heads/mg_head_v4_release.py", line 612, in loss
    iou_pred_loss = iou_pred_loss.sum() / batch_size
RuntimeError: CUDA error: invalid configuration argument

it gives me the same error
The error is in the sum() function and yes I uses IOU3D_CUDA  from this repo: GitHub - sshaoshuai/PointRCNN: PointRCNN: 3D Object Proposal Generation and Detection from Point Cloud, CVPR 2019. 1