RankSRGAN icon indicating copy to clipboard operation
RankSRGAN copied to clipboard

Published 20 hours ago • verified publisher icon XPixelGroup

training time and number of GPUs

Open Marshall-yao opened this issue 5 years ago • 2 comments

Hi, thanks for your wonderful work and opening source. Could you please tell me how long did you train the model , the kind of GPU and number of GPUs?

Best regards

Marshall-yao avatar Aug 22 '19 06:08 Marshall-yao

Thanks for your interest in our work! It takes about 55 hours (total 500 K iterations) to train the RankSRGAN using 4 Titan Xp GPUs. The training time is slower than standard SRGAN. The reason is that the patch size of standard SRGAN is set to 128, due to Ranker needs a larger size to predict the ranking score, we set the patch size to 296.

Best regards

WenlongZhang0517 avatar Aug 22 '19 09:08 WenlongZhang0517

@WenlongZhang0724 Thanks very much for your patient reply. Yes, due to larger patch size, training time is longer.

Marshall-yao avatar Aug 22 '19 13:08 Marshall-yao