Swin-Transformer-Object-Detection 訓練

以下是使用Swin Transformer Object Detection 訓練自己的資料集可能發生的問題

在single gpu 可能發生的問題：

apex可能會無法正常運作，以cascade_mask_rcnn_swin_tiny_patch4_window7_mstrain_480-800_giou_4conv1f_adamw_1x_coco.py為例

須將runner = dict(type='EpochBasedRunnerAmp', max_epochs=12)修改為EpochBasedRunner

並且將

optimizer_config = dict(

     type="DistOptimizerHook",

     update_interval=1,

     grad_clip=None,

     coalesce=True,

     bucket_size_mb=-1,

     use_fp16=True,

 )

comment起來

再來norm_cfg=dict(type='SyncBN', requires_grad=True)，須將SyncBN替換為BN


留言

張貼留言

這個網誌中的熱門文章

在windows 10上安裝mmdetection

 測試環境 cuda=11.1.0 GPU=3070 首先使用conda虛擬環境 安裝 conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cudatoolkit=11.1 -c pytorch -c conda-forge 設定MSVS C++ Compiler(只能用VS2019) 添加 `C:\Program Files (x86)\Microsoft Visual Studio\2019\Community\VC\Tools\MSVC\14.27.29110\bin\Hostx86\x64` 到 `PATH` 安裝mmcv時可能會有問題，可以測試修改 #!/bin/bash SITE = /mnt/c/Users/thomas.jungblut/AppData/Local/Programs/Python/Python37/Lib/site-packages sed -i.bak -e 's/CONSTEXPR_EXCEPT_WIN_CUDA/const/g' " ${ SITE } /torch/include/torch/csrc/jit/api/module.h" sed -i.bak -e 's/return \*(this->value)/return \*((type\*)this->value)/g' " ${ SITE } /torch/include/pybind11/cast.h" sed -i.bak '/static constexpr Symbol Kind/d' " ${ SITE } /torch/include/torch/csrc/jit/ir/ir.h" 修改SITE為自己的環境的路徑，這裡可利用linux subsystems執行 /mnt/c對應到本機C: 安裝mmcv git clone https://github.com/open-mmlab/mmcv.git 注意：要使用conda powershell才可以執行 cd mmcv pip install -r requirements.txt $env:M...
閱讀完整內容