Allow for GPU usage on M1 mac

sanghyun-son / EDSR-PyTorch

PyTorch version of the paper 'Enhanced Deep Residual Networks for Single Image Super-Resolution' (CVPRW 2017)

MIT License

2.43k stars 670 forks source link

Closed mradassaad closed 1 year ago

mradassaad commented 1 year ago

Checks for either CUDA or MPS device for GPU use
Fixes an issue with the multiprocessing library not being able to pickle bg_process

sanghyun-son commented 1 year ago

Thank you for the contribution!