uzh-rpg / rpg_e2vid

Code for the paper "High Speed and High Dynamic Range Video with an Event Camera" (T-PAMI, 2019).
http://rpg.ifi.uzh.ch/E2VID.html
GNU General Public License v3.0
353 stars 128 forks source link

Running the model on a CPU #23

Open UtharaKeerthan opened 2 years ago

UtharaKeerthan commented 2 years ago

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 commented 2 years ago

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