search

sanghyun-son / EDSR-PyTorch

PyTorch version of the paper 'Enhanced Deep Residual Networks for Single Image Super-Resolution' (CVPRW 2017)
MIT License
2.41k stars 666 forks source link

question about GPU memory used #311

Closed UESTCrookieLI closed 2 years ago

UESTCrookieLI commented 3 years ago

hello.thanks your great work EDSR! I want to train the model,but i wonder why my program just use 1271MB GPU memory. I run the conmand "python main.py --model EDSR --scale 2 --patch_size 96 --save edsr_baseline_x2 --reset --ext bin" to train the EDSR.

could your help me?