testing model on a cpu-only machine

aatifjiwani / rgb-footprint-extract

A Semantic Segmentation Network for Urban-Scale Building Footprint Extraction Using RGB Satellite Imagery

https://arxiv.org/abs/2104.01263

MIT License

65 stars 22 forks source link

testing model on a cpu-only machine #16

Closed teresalisanti closed 2 years ago

teresalisanti commented 2 years ago

Hi :) i run the infer_deeplab.sh setting --no-cuda but i get this error:

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.

Can you help me? Thank you

I added map_location=torch.device('cpu') in torch.load in loader.py