如何训练?

作者: LiJichen0114创建于 2025年1月7日更新于 2026年1月17日

你好,非常感谢您的工作。我很想知道如何从头开始在 ImageNet 数据集上训练 ViT 模型。我没有看到用于训练的 Python 文件。谢谢!

内容来源: lucidrains/vit-pytorch