Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
wangsen
Swin Transformer
Commits
024c70ea
Commit
024c70ea
authored
Aug 22, 2024
by
root
Browse files
add readme.md
parents
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
58 additions
and
0 deletions
+58
-0
README.md
README.md
+58
-0
No files found.
README.md
0 → 100644
View file @
024c70ea
# 数据集
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
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment