在 DiT-S/2 上重现问题

你好,感谢你的伟大工作! 我尝试使用这个仓库在 256×256 分辨率下重现 DiT-S/2 的性能。我的训练设置遵循了 README 中提供的配置以及训练脚本中的默认参数,即: • 全局批量大小 = 256 • 学习率 = 1e-4 然而,在完成 400 万步后,模型的 FID 仍然远远低于论文中报告的 68.4。即使在完成 4000 万步后,我也只能在测试集上实现 FID 为 79.59。