cszn / KAIR

Image Restoration Toolbox (PyTorch). Training and testing codes for DPIR, USRNet, DnCNN, FFDNet, SRMD, DPSR, BSRGAN, SwinIR
https://cszn.github.io/
MIT License
2.95k stars 630 forks source link

DPSR training problem #11

Open wenfan-wenfan opened 4 years ago

wenfan-wenfan commented 4 years ago

Thanks for @cszn contribution, DPSR is an amazing job, so I give author a star to say thanks! But in my training,It takes a long time. When I checked, I found that my CPU usage was very high and GPU usage was very low(such as image1 and image2),So I wanted to ask: 1.How long does a DPSR training normally take? 2.Have you ever experienced high CPU usage and low GPU usage before?How do you solve the problem of low GPU usage and high CPU usage?

My email address is 861847316@qq.com,I look forward to further discussion with you. image 1: https://imgchr.com/i/8shnK0 image 2 : https://imgchr.com/i/8s4wyq

cszn commented 4 years ago

Change batch size https://github.com/cszn/KAIR/blob/6d4ff120435ca2e0f8d0cdfc8aa7e3c59f24d281/options/train_dpsr.json#L28