The retrained result on COCO

tobran / DF-GAN

[CVPR2022 oral] A Simple and Effective Baseline for Text-to-Image Synthesis

Other

297 stars 67 forks source link

The retrained result on COCO #5

Closed shangshng closed 3 years ago

shangshng commented 3 years ago

Hello,

Thank you for your works. I have some problems when retraining the model on COCO. The retraining FID is 56.46 using the default setting of coco.yml (batchsize = 24, nf = 32), and the enviroment is py36, pytorch1.5 with a 2080Ti. However, the provided pretrained model achieves 26.68 FID with the same setting. In addition, have you tried Inception Score on COCO?

I'll really appreciate it if you could give me some advice.

shangshng commented 3 years ago

I've tried batchsize = 48 using a 3090, and get the FID = 29.68. So I guess the pretrained model is trained on V100 with a lager batchsize, right ?

Looking forward to your reply.