Commit f197d0a5 authored by chenych's avatar chenych
Browse files

change scnet model in README

parent 44387c65
...@@ -64,24 +64,25 @@ pip install -r requirements.txt ...@@ -64,24 +64,25 @@ pip install -r requirements.txt
``` ```
## 训练 ## 训练
下载预训练模型放在checkpoint目录下:
``` 下载预训练模型[ViT-B_16.npz](http://113.200.138.88:18080/aimodels/findsource-dependency/ViT-B_16.git)放在checkpoint目录下。
wget https://storage.googleapis.com/vit_models/imagenet21k/ViT-B_16.npz
```
### 单机单卡 ### 单机单卡
``` ```
export HIP_VISIBLE_DEVICES=0 export HIP_VISIBLE_DEVICES=0
python3 -m torch.distributed.launch --nproc_per_node=1 --use-env 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 --use-env 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 --use-env 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 --use-env 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 ## result
![1695381570003](image/README/1695381570003.png) ![1695381570003](image/README/1695381570003.png)
### 精度 ### 精度
测试数据使用的是cifar10,使用的加速卡是DCU Z100L。 测试数据使用的是cifar10,使用的加速卡是DCU Z100L。
......
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