Commit 6cf8a836 authored by suily's avatar suily
Browse files

Update README.md

parent f0355acd
...@@ -28,7 +28,7 @@ ViT主要包括patch embeding、transformer encoder、MLP head三部分:以图 ...@@ -28,7 +28,7 @@ ViT主要包括patch embeding、transformer encoder、MLP head三部分:以图
## 代码改动说明 ## 代码改动说明
ps:仓库中是改动后的代码,不需再次修改 ps:仓库中是改动后的代码,不需再次修改
``` ```
1、vision_transformer/vit_jax/train.py修改train_and_evaluate函数: 1、vision_transformer/vit_jax/train.py修改train_and_evaluate函数:
input_pipeline.get_datasets(config) # TODO:添加,解决直接get_dataset_info加载信息失败的问题 input_pipeline.get_datasets(config) # TODO:添加,解决直接get_dataset_info加载信息失败的问题
``` ```
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment