tobran / DF-GAN

[CVPR2022 oral] A Simple and Effective Baseline for Text-to-Image Synthesis
Other
297 stars 67 forks source link

如何复现 coco dataset 上的 fid 指标 #32

Open JamesSand opened 1 year ago

JamesSand commented 1 year ago

作者您好。我看到 paper 中提供的最终 model 在 coco dataset 上的 fid 是 19.32,但是我下载了您 readme 里边提供的 checkpoint,并运行 bash scripts/calc_fid.sh ./cfg/coco.yml 来测量 coco val2014 上的 fid。但是我得到的结果是 15.59

我是完全按照 readme 里边的顺序一步一步 setup 的。唯一的区别是 codebase 使用的 pytorch 和 torchvision 的版本不 support 我使用的 3090 所以我将 torch 升级到 1.13.1 torchvision 升级到 0.14 。这样测量出来的结果是 15.59

我不知道我哪一步出了问题,我只是升级了 torch 的版本适配到我的机器上。

期待您的解答