Keras-SRGAN icon indicating copy to clipboard operation
Keras-SRGAN copied to clipboard

Memory Error when training the model

Open mustaeenqazi opened this issue 4 years ago • 7 comments

Hi Deepak, Thanks for the amazing code, a small issue, when i start training the model with your simplified version of code, the Ram gets exhausted. although i have 12GB ram on my google colab, I tried reducing the size of images but no results

mustaeenqazi avatar Jul 05 '20 15:07 mustaeenqazi

have you solve the problem?pls

hongyanghu avatar Aug 15 '20 06:08 hongyanghu

I had the same problem. What batch size are you using? Reducing that should help. I also halved the number of neurons in the dense layers and number of filters in the conv layers, which of course will impact performance.

rcbrigg avatar Aug 24 '20 19:08 rcbrigg

The problem is solved by using small batch size and lower number of images. but you can also try to upgrade your memory.

mustaeenqazi avatar Sep 04 '20 22:09 mustaeenqazi

Dear Sir, Brilliant work , thank you for sharing your code.. Please, according to your GPU version what the time did you take to finish training your network ?? I'm waiting for your fast reply..

BassantTolba1234 avatar Nov 18 '20 21:11 BassantTolba1234

Dear Sir, Brilliant work , thank you for sharing your code.. Please, according to your GPU version what the time did you take to finish training your network ?? I'm waiting for your fast reply..

BassantTolba1234 avatar Nov 18 '20 21:11 BassantTolba1234

Please all, I need the code of implementation this part the part is

{The SRResNet networks were trained with a learning rate of 10−4 and 106 update iterations. We employed the trained MSE-based SRResNet network as initialization for the generator when training the actual GAN to avoid undesired local optima.{

BassantTolba1234 avatar Nov 28 '20 09:11 BassantTolba1234

Could you please tell me which size of your image before super resolution?

MuterDear avatar Dec 08 '21 07:12 MuterDear