Update README.md

README.md

@@ -24,7 +24,12 @@ ViT主要包括patch embeding、transformer encoder、MLP head三部分：以图
 <div align=center>
     <img src="./doc/vit.png"/>
 </div>
-
+## 代码改动说明
+ps：仓库中是改动后的代码，不需再次修改
+```
+1、vision_transformer/vit_jax/train.py。修改train_and_evaluate函数：
+input_pipeline.get_datasets(config) # TODO：添加，解决直接get_dataset_info加载信息失败的问题
+```
 ## 环境配置
 ### Docker（方法一）
 ```