I can train this model using one gpu?

swz30 / MIRNet

[ECCV 2020] Learning Enriched Features for Real Image Restoration and Enhancement. SOTA results for image denoising, super-resolution, and image enhancement.

Other

667 stars 96 forks source link

I can train this model using one gpu? #7

Closed AselSiriwardena closed 3 years ago

AselSiriwardena commented 3 years ago

Hi, I have followed your which is awesome! I wanted to train this model with some additional data. will it be possible to do it on a gtx 1660ti or a rtx 2060 laptop GPU?

I'm still a student. That's the best gpu I can afford right now.

Thanks!

adityac8 commented 3 years ago

HI @AselSiriwardena Thank you for your interest in our work. You can have a look at #5 for details on adjusting parameters, such as batch_size and n_feat for your GPU of 6GB memory. Please note that changing these parameters would result in a trade-off between training time and performance. Thanks

AselSiriwardena commented 3 years ago

Thank you so much for your reply! I have one more question.

In #5 you have mentioned that you used two NVIDIA Tesla V100 GPUs. are you using a cloud service?