RuntimeError: CUDA out of memory.

[rezraz1]

Hi, I tried to run the code with trained models (EDSR) and got an error about CUDA and GPU that is given below :

RuntimeError: CUDA out of memory. Tried to allocate 2.64 GiB (GPU 0; 4.00 GiB to
tal capacity; 1.15 GiB already allocated; 2.21 GiB free; 1.19 GiB reserved in to
tal by PyTorch)

Apparently this error occurs due to the GPU, now I want to know if there is a possibility for the code to run on the CPU? How can the code be changed?
(If we consider that there is no memory problem.)

@sanghyun-son
Thank you very much for your guidance