Closed Qiulin-W closed 5 months ago
Hi! Thanks for your compliments. Training with only CelebA-HQ is quite a dangerous choice. The dataset bias can be affected to total loss. In your loss graph, reconstruction loss is very high. Maybe you should try to increase the coefficient of reconstruction loss if you wanna train with only CelebA-HQ.
Be careful about overfitting.
Hi! Thanks for your compliments. Training with only CelebA-HQ is quite a dangerous choice. The dataset bias can be affected to total loss. In your loss graph, reconstruction loss is very high. Maybe you should try to increase the coefficient of reconstruction loss if you wanna train with only CelebA-HQ.
Be careful about overfitting.
Thank you so much for your reply!
By the way, what is the expected value of reconstruction loss of your well-trained model? If possible, could you share your loss graph as a reference? Did you set grad_clip to zero?
Could you please tell me how long does it take to finish the training?
Could you please tell me how long does it take to finish the training?
For celebHQ dataset only, it takes 2-3 hours for an epoch.
Could you please tell me how long does it take to finish the training?
For celebHQ dataset only, it takes 2-3 hours for an epoch.
can you benefit from multi-GPU training? does it really increase the training speed?
已经过去四年了,请问下你能分享下arcface.pth文件吗,链接已经失效了,感激不尽
Thanks for the great work!
When I try to train the AEI-Net with 30k images from celebHQ dataset using 6 P40 32G GPUs, I got the training curve as below:
![Screenshot from 2020-11-23 11-41-41](https://user-images.githubusercontent.com/37483535/99927877-e0c57480-2d81-11eb-9be8-a8f6f7c15ba8.png)
All the other setting are set by default and the generated swap faces are also weird:
![output2](https://user-images.githubusercontent.com/37483535/99927946-1e2a0200-2d82-11eb-9b9e-49c7dd7cd372.png)
Should I continue training or any sugguestions? Thanks in advance!