the time of model_restoration.cuda() is too long and failed in the end.

[mrchasezhang]

Hi, Dear author. it is my honor to learn your work. The MPRNet is very great, when I run your code of Deblurring, I met a problem. When I run the train.py, the speed of loading code to GPU is very slow. The results shows that model_restoration.cuda() is very sluggish after debuging. Can u tell me how to deal with it?