代码复现报错An Image is Worth 16x16 Words: Transformers for Image Recognition at Scale

Question

复现论文里的代码。
在autodl上创建了一个虚拟环境。
下载代码到虚拟环境
用模型里的配置文件配置了环境
下载了cifar10数据集。
用指令运行(myenv) root@autodl-container-c78a4bb56d-cf0bddef:/workspace/vit_project/vision_transformer# python -m vit_jax.main --workdir=/tmp/vit-$(date +%s)
--config=$(pwd)/vit_jax/configs/vit.py:b16,cifar10
--config.pretrained_dir='gs://vit_models/imagenet21k'
出现以下报错
KeyError: "Trying to access splits['train'] but splits is empty. This likely indicates the dataset has not been generated yet."
求问如何解决？

Answer 1

阿里嘎多学长整理AIGC生成，因移动端显示问题导致当前答案未能完全显示，请使用PC端查看更加详细的解答过程

问题解答

你想复现论文《An Image is Worth 16x16 Words: Transformers for Image Recognition at Scale》中的代码，但是遇到了报错。

根据你的描述，你已经创建了虚拟环境、下载了代码、配置了环境和数据集，但是仍然报错。

可能的解决方案：

1. 检查代码中的依赖项是否正确，确保所有依赖项都已经安装。
2. 检查模型配置文件是否正确，确保模型的架构和 hyperparameters 都正确。
3. 检查数据集是否正确，确保数据集的格式和路径正确。
4. 检查运行命令是否正确，确保命令中的路径和参数正确。

如果以上解决方案都不能解决问题，可以提供更多的错误信息和代码片段，以便更好地帮助你解决问题。

代码片段

如果你想提供代码片段，可以提供以下几个部分的代码：

• 模型配置文件
• 数据集加载代码
• 运行命令

这些代码片段可以帮助我更好地理解你的问题，并提供更加具体的解决方案。