Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
ModelZoo
vision_transformer_jax
Commits
af33744e
Commit
af33744e
authored
Apr 17, 2025
by
chenzk
Browse files
Update url.md
parent
8cbdc81f
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
4 additions
and
5 deletions
+4
-5
README.md
README.md
+4
-5
No files found.
README.md
View file @
af33744e
...
@@ -112,7 +112,7 @@ vim /usr/local/lib/python3.10/site-packages/tensorflow_datasets/core/utils/gcs_u
...
@@ -112,7 +112,7 @@ vim /usr/local/lib/python3.10/site-packages/tensorflow_datasets/core/utils/gcs_u
│ └── label.labels.txt
│ └── label.labels.txt
```
```
### 推理数据集
### 推理数据集
推理所用图片和文件可根据
[
scnet
](
http://113.200.138.88:18080/aidatasets/project-dependency/vision_transformer_jax
)
或
以下代码进行下载:
推理所用图片和文件可根据以下代码进行下载:
```
```
# ./dataset是存储地址,可自订
# ./dataset是存储地址,可自订
wget https://storage.googleapis.com/bit_models/ilsvrc2012_wordnet_lemmas.txt -P ./dataset
wget https://storage.googleapis.com/bit_models/ilsvrc2012_wordnet_lemmas.txt -P ./dataset
...
@@ -125,7 +125,7 @@ wget https://picsum.photos/384 -O ./dataset/picsum.jpg # 将图片调整为384
...
@@ -125,7 +125,7 @@ wget https://picsum.photos/384 -O ./dataset/picsum.jpg # 将图片调整为384
│ └── picsum.jpg
│ └── picsum.jpg
```
```
## 训练
## 训练
检查点可通过
[
scnet
](
http://113.200.138.88:18080/aimodels/findsource-dependency/vision_transformer_jax/-/tree/master/imagenet21k?ref_type=heads
)
或
以下方式进行下载:
检查点可通过以下方式进行下载:
```
```
cd /your_code_path/vision_transformer/test_result # test_result为检查点下载地址,可自订
cd /your_code_path/vision_transformer/test_result # test_result为检查点下载地址,可自订
wget https://storage.googleapis.com/vit_models/imagenet21k/ViT-B_16.npz
wget https://storage.googleapis.com/vit_models/imagenet21k/ViT-B_16.npz
...
@@ -147,7 +147,7 @@ sh test.sh
...
@@ -147,7 +147,7 @@ sh test.sh
# config.optim_dtype='bfloat16' # 精度
# config.optim_dtype='bfloat16' # 精度
```
```
## 推理
## 推理
检查点可通过
[
scnet
](
http://113.200.138.88:18080/aimodels/findsource-dependency/vision_transformer_jax/-/tree/master/imagenet21k+imagenet2012?ref_type=heads
)
或
以下方式进行下载:
检查点可通过以下方式进行下载:
```
```
cd /your_code_path/vision_transformer/test_result # test_result为检查点下载地址,可自订
cd /your_code_path/vision_transformer/test_result # test_result为检查点下载地址,可自订
wget https://storage.googleapis.com/vit_models/imagenet21k+imagenet2012/ViT-B_16.npz -O ViT-B_16_imagenet2012.npz
wget https://storage.googleapis.com/vit_models/imagenet21k+imagenet2012/ViT-B_16.npz -O ViT-B_16_imagenet2012.npz
...
@@ -234,10 +234,9 @@ k100*1(1270Mhz,64G,dtk24.04.1):
...
@@ -234,10 +234,9 @@ k100*1(1270Mhz,64G,dtk24.04.1):
### 热点应用行业
### 热点应用行业
`制造,电商,医疗,广媒,教育`
`制造,电商,医疗,广媒,教育`
## 预训练权重
## 预训练权重
-
http://113.200.138.88:18080/aimodels/findsource-dependency/vision_transformer_jax
-
https://console.cloud.google.com/storage/browser/vit_models/imagenet21k/ (微调用)
-
https://console.cloud.google.com/storage/browser/vit_models/imagenet21k/ (微调用)
https://console.cloud.google.com/storage/browser/vit_models/imagenet21k+imagenet2012/ (推理用)
-
https://console.cloud.google.com/storage/browser/vit_models/imagenet21k+imagenet2012/ (推理用)
## 源码仓库及问题反馈
## 源码仓库及问题反馈
-
https://developer.hpccube.com/codes/modelzoo/vision_transformer_jax
-
https://developer.hpccube.com/codes/modelzoo/vision_transformer_jax
## 参考资料
## 参考资料
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment