Update README.md

README.md

@@ -24,12 +24,14 @@ 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（方法一）
 ```