Any flag to run the whole model on CPU?

ZhihaoHu / PyTorchVideoCompression

PyTorch implementation and benchmark of Video Compression

MIT License

321 stars 71 forks source link

Any flag to run the whole model on CPU? #5

Open SagarB-97 opened 2 years ago

SagarB-97 commented 2 years ago

I'm trying to test on CPU, but I've had to make lots of changes to the code. (Remove DataParallel, .cuda() etc.) Is there a flag that can allow me to easily switch between CPU and GPU?

ChaosAdmStudent commented 6 months ago

Were you able to do this?