#110·DiT

数据集?

作者: IamMI创建于 2025年3月6日更新于 2025年3月6日

目前我正在尝试找到训练数据集。虽然论文中指定了 ImageNet,但是在下载 ILSVRC 2012 后,我注意到标签与代码中的标签不一致!例如,在代码中,我设置 class_labels=[15] 以生成一张鸟的图像,但在 ILSVRC 2012 数据集中,我发现了一个包含海狮图像的数据集!这让我对用于训练模型的具体数据集产生了好奇。

内容来源: facebookresearch/DiT