Test scan fix for non-cuda devices

PRBonn / make_it_dense

Make it Dense: Self-Supervised Geometric Scan Completion of Sparse 3D LiDAR Scans in Large Outdoor Environments

https://www.ipb.uni-bonn.de/wp-content/papercite-data/pdf/vizzo2022ral-iros.pdf

133 stars 12 forks source link

Test scan fix for non-cuda devices #9

Closed vnmsklnk closed 1 year ago

vnmsklnk commented 1 year ago

This PR fixes bug with running test scan on non-cuda devices:

RuntimeError: Attempting to deserialize object on a CUDA device but torch.cuda.is_available() is False. If you are running on a CPU-only machine, please use torch.load with map_location=torch.device('cpu') to map your storages to the CPU.