Commit 08b7113e authored by sunxx1's avatar sunxx1
Browse files

updata readme

parent 19ee37d6
# Vision Transformer(ViT) # ViT
## 论文 ## 论文
...@@ -61,6 +61,17 @@ cifar10 ...@@ -61,6 +61,17 @@ cifar10
链接:https://pan.baidu.com/s/1ZFMQVBGQZI6UWZKJcTYPAQ?pwd=fq3l 提取码:fq3l 链接:https://pan.baidu.com/s/1ZFMQVBGQZI6UWZKJcTYPAQ?pwd=fq3l 提取码:fq3l
```
├── batches.meta
├── data_batch_1
├── data_batch_2
├── data_batch_3
├── data_batch_4
├── data_batch_5
├── readme.html
└── test_batch
```
## 训练 ## 训练
下载预训练模型放在checkpoint目录下: 下载预训练模型放在checkpoint目录下:
...@@ -78,7 +89,8 @@ wget https://storage.googleapis.com/vit_models/imagenet21k/ViT-B_16.npz ...@@ -78,7 +89,8 @@ wget https://storage.googleapis.com/vit_models/imagenet21k/ViT-B_16.npz
``` ```
python3 -m torch.distributed.launch --nproc_per_node=8 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=8 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
``` ```
## result
![1695381570003](image/README/1695381570003.png)
## 精度 ## 精度
测试数据使用的是cifar10,使用的加速卡是DCU Z100L。 测试数据使用的是cifar10,使用的加速卡是DCU Z100L。
......
# 模型唯一标识
modelCode =105
# 模型名称 # 模型名称
modelName=ViT-PyTorch modelName=vit_pytorch
# 模型描述 # 模型描述
modelDescription=ViT是一种新的基于Transformer应用于计算机视觉领域的神经网络模型,基于PyTorch实现测试 modelDescription=ViT是一种新的基于Transformer应用于计算机视觉领域的神经网络模型,基于PyTorch实现测试
# 应用场景 # 应用场景
appScenario=训练图像分类,目标检测 appScenario=训练,图像分类,制造,交通,网安
# 框架类型 # 框架类型
frameType=PyTorch frameType=pytorch
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