赞
踩
同MMdetection环境配置一样,可参考我上一篇文章
下载仓库时,将git clone的地址换成Swin-Transformer的仓库地址
- git clone https://github.com/SwinTransformer/Swin-Transformer-Object-Detection.git
- cd Swin-Transformer-Object-Detection
- git clone https://github.com/open-mmlab/mmcv.git
- cd mmcv
- MMCV_WITH_OPS=1 pip install -e . # package mmcv-full will be installed after this step
- cd ..
- 注意-e 后面一点不要漏了
- pip install -r requirements/build.txt
- pip install -v -e . # or "python setup.py develop"
如果安装,输入
- git clone https://github.com/NVIDIA/apex
- cd apex
- pip install -v --disable-pip-version-check --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" ./
则需要在配置文件末尾,将下述代码注释
- # do not use mmdet version fp16
- fp16 = None
- optimizer_config = dict(
- type="DistOptimizerHook",
- update_interval=1,
- grad_clip=None,
- coalesce=True,
- bucket_size_mb=-1,
- use_fp16=True,
- )
并将
- runner = dict(type='EpochBasedRunnerAmp',max_epochs=36)
- 改成
- runner = dict(type='EpochBasedRunner',max_epochs=36)
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。