search

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

stuck at g_optim.step() #39

Open dcnstrctn opened 1 year ago

dcnstrctn commented 1 year ago

Hello. When I run train.py on my own 10-shot images (source domain: FFHQ), the training always stuck at g_optim.step() in "if g_regularize:". It always takes about half an hour to process this line. My device is NVIDIA 3090 with Cuda 11.4. How can I solve this issue?