Running the model on a CPU

[UtharaKeerthan]

Hi ,
Thanks for the model and the repository.
Is there an option to run the model with only cpu or the code is intended to run on GPUs by default? Could you please provide your feedback ?

I set the default option of 'use_gpu = False' but still I get the following error, when I run the run_reconstruction.py script.

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.

[e-sha]

Hello,
I think this Pull request should help you with the issue.