How to train high-resolution models with limited GPU resources ?

WisconsinAIVision / few-shot-gan-adaptation

[CVPR '21] Official repository for Few-shot Image Generation via Cross-domain Correspondence

https://utkarshojha.github.io/few-shot-gan-adaptation/

Other

291 stars 47 forks source link

How to train high-resolution models with limited GPU resources ? #37

Open li-car-fei opened 2 years ago

li-car-fei commented 2 years ago

I only have 2 2080 GPUs, but I want to train a generator with a resolution of 1028*1028. How can I adjust the complex loss and parameters so that the model can be trained?

Since my generator is already pretrained, I guess I can fine-tune it by this method.