Skip to content

GitLab

Commit 84d99b46 authored by sunxx1's avatar sunxx1
Browse files

修改redame

parent 59907253
Hide whitespace changes
Inline Side-by-side
Showing with 8 additions and 4 deletions
README.md
View file @ 84d99b46
...@@ -35,12 +35,16 @@ wget https://storage.googleapis.com/vit_models/imagenet21k/ViT-B_16.npz ...@@ -35,12 +35,16 @@ wget https://storage.googleapis.com/vit_models/imagenet21k/ViT-B_16.npz
export HIP_VISIBLE_DEVICES=3 export HIP_VISIBLE_DEVICES=3
python3 -m torch.distributed.launch --nproc_per_node=1 train.py --name cifar10-100_500 --dataset cifar10 --model_type ViT-B_16 --pretrained_dir checkpoint/ViT-B_16.npz --train_batch_size 64 --num_steps 500 python3 -m torch.distributed.launch --nproc_per_node=1 train.py --name cifar10-100_500 --dataset cifar10 --model_type ViT-B_16 --pretrained_dir checkpoint/ViT-B_16.npz --train_batch_size 64 --num_steps 500
## 性能和准确率数据 ## 准确率数据
测试数据使用的是cifar10，使用的加速卡是DCU Z100L。 测试数据使用的是cifar10，使用的加速卡是DCU Z100L。
| 卡数 | 性能 | 精度 | | 卡数 | 精度 |
| :------: | :------: | :------: | | :------: | :------: |
| 1 | 67.84 samples/s | Best Accuracy=0.3051 | | 1 | Best Accuracy=0.3051 |
### 历史版本
https://developer.hpccube.com/codes/modelzoo/vit-pytorch
### 参考 ### 参考
https://github.com/jeonsworld/ViT-pytorch https://github.com/jeonsworld/ViT-pytorch
\ No newline at end of file
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