Run on CPU

[NaeemKhan333]

How to run this code on CPU machine.please guide me about this.Thanks

[zjjzjj123]

you can modify '.cuda()' this code and modify SPPE/src/main_fast_inference.py code 'torch.load('.....du.....th')' ->'torch.load('.....du.....th',map_location='cpu)'
i think is ok on video_demo.py

[zjjzjj123]

@NaeemKhan333

[Ashish-Surve]

@zjjzjj123 - what do you mean by modify '.cuda()'. Am I supposed to just remove it from dataloader.py? for video.

[i-m-arin]

There are a lot of files where CUDA is used. Changing every one of them manually is quite difficult. Is there any other way?