Evaluation without CUDA

[Shan-handsome]

Thank you for your nice implementation.
I tried running evaluation without cuda as the command below:
python evaluate.py --ckpt pretrained/epoch_160.pth --data_root kitti/ --no_cuda

This command ends up with an error in the function nms_gpu():
TypeError: nms_gpu(): incompatible function arguments. The following argument types are supported:
1. (arg0: at::Tensor, arg1: at::Tensor, arg2: float, arg3: int) -> int

Is there a cpu version of nms? Or is there any way can fix this error?
Thanks.