Skip to content

GitLab

Commit 983e779c authored by suily's avatar suily
Browse files

添加推理结果

parent 568b9c2f
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
README.md
View file @ 983e779c
...@@ -147,7 +147,7 @@ wget https://storage.googleapis.com/vit_models/imagenet21k+imagenet2012/ViT-L_16 ...@@ -147,7 +147,7 @@ wget https://storage.googleapis.com/vit_models/imagenet21k+imagenet2012/ViT-L_16
``` ```
``` ```
cd /your_code_path/vision_transformer cd /your_code_path/vision_transformer
python test.py python test.py # 文件内可修改模型目录和数据集目录
``` ```
## result ## result
测试图为： 测试图为：
......
test.py
View file @ 983e779c
...@@ -26,7 +26,7 @@ if not (jax_test=='gpu'): ...@@ -26,7 +26,7 @@ if not (jax_test=='gpu'):
exit() exit()
'''指定模型''' '''指定模型'''
model_name = 'ViT-L_16' #@param ["ViT-B_32", "Mixer-B_16"] model_name = 'ViT-B_16' #@param ["ViT-B_32", "Mixer-B_16"]
pretrained_path=f'./test_result/{model_name}.npz' pretrained_path=f'./test_result/{model_name}.npz'
model_path=f'./test_result/{model_name}_imagenet2012.npz' model_path=f'./test_result/{model_name}_imagenet2012.npz'
......
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