Skip to content

Commit

Permalink
add results
Browse files Browse the repository at this point in the history
  • Loading branch information
raoyongming committed Jun 6, 2021
1 parent 50c397b commit 5a189e8
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,7 @@ Our code is based on [pytorch-image-models](https://github.com/rwightman/pytorch

## Model Zoo

We provide our DynamicViT models pretrained on ImageNet
We provide our DynamicViT models pretrained on ImageNet:
| name | arch | rho | acc@1 | acc@5 | GFLOPs | url |
| --- | --- | --- | --- | --- | --- | --- |
| DynamicViT-256/0.7 | ```deit_256``` | 0.7 | 76.532 | 93.118 | 1.3 | [Google Drive](https://drive.google.com/file/d/1fpdTNRZtGOW25UwOadj1iUdjqmu88WkO/view?usp=sharing) / [Tsinghua Cloud](https://cloud.tsinghua.edu.cn/f/ebda4114758f44d78bc0/?dl=1) |
Expand Down Expand Up @@ -90,6 +90,8 @@ LV-ViT-M
python -m torch.distributed.launch --nproc_per_node=8 --use_env main_dynamic_vit.py --output_dir logs/dynamic-vit_lvvit-m --arch lvvit_m --input-size 224 --batch-size 48 --data-path /path/to/ILSVRC2012/ --epochs 30 --dist-eval --distill --base_rate 0.7
```

You can train models with different keeping ratio by adjusting ```base_rate```. DynamicViT can also achieve comparable performance with only 15 epochs training (around 0.1% lower accuracy).

## License
MIT License

Expand Down

0 comments on commit 5a189e8

Please sign in to comment.