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
```
## 训练
下载预训练模型放在checkpoint目录下:
```
wget https://storage.googleapis.com/vit_models/imagenet21k/ViT-B_16.npz
```
下载预训练模型[ViT-B_16.npz](http://113.200.138.88:18080/aimodels/findsource-dependency/ViT-B_16.git)放在checkpoint目录下。
### 单机单卡
```
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=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
![1695381570003](image/README/1695381570003.png)
### 精度
测试数据使用的是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