支持GPU计算

[xiaohangguo]

你好，很棒的工作，请问支持gpu运行只需要改run_kitti部分，使得模型加载在gpu上吗？
checkpoint = torch.load(args.checkpoint, map_location=torch.device('cpu'))

[lturing]

nms需要切换成gpu版，安装运行 bash setup.py
以及nms的实现
抱歉哈，因为我笔记本上没gpu，所以特意将gpu版改为cpu版。
你可以先运行代码，根据报错信息逐步修改。或者直接运行官方代码