Commit 024c70ea authored by root's avatar root
Browse files

add readme.md

parents
# 数据集
imagenet2012
```
apt-get install git-lfs
git lfs clone http://113.200.138.88:18080/aidatasets/project-dependency/imagenet-2012.git
```
# 下载模型
```
模型:swinv2-large-w24_in21k-pre_3rdparty_in1k-384px
wget https://github.com/open-mmlab/mmpretrain.git
```
# 环境搭建
```
cd mmpretrain
pip install openmim
mim install -e . -i https://pypi.tuna.tsinghua.edu.cn/simple
```
# 训练
## 单卡执行
```
# 修改
vim configs/_base_/models/swin_transformer_v2/large_384.py
vim configs/_base_/datasets/imagenet_bs64_swin_384.py
vim configs/_base_/schedules/imagenet_bs1024_adamw_swin.py
vim configs/_base_/default_runtime.py
python tools/train.py configs/swin_transformer_v2/swinv2-large-w24_in21k-pre_16xb64_in1k-384px.py --amp
```
## 多卡执行
```
bash ./tools/dist_train.sh configs/swin_transformer_v2/swinv2-large-w24_in21k-pre_16xb64_in1k-384px.py 8 --amp
```
# 推理
# 参考文档
https://github.com/open-mmlab/mmpretrain.git
https://github.com/open-mmlab/mmpretrain/tree/main/configs/swin_transformer_v2
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