sangwoomo / FreezeD

Freeze the Discriminator: a Simple Baseline for Fine-Tuning GANs (CVPRW 2020)
https://arxiv.org/abs/2002.10964
285 stars 37 forks source link

Cuda out of memory #11

Open MotiBaadror opened 2 years ago

MotiBaadror commented 2 years ago

Hi,

Thank you for organizing the code. I just want to know the minimum gpu requirement and time the model will take to finetune after we have that resource. Since with 8 GB GPU memory even batch_size =1 ( in finetune.py ) , gives the cuda out of memory error.