2023年11月29日发(作者:)
Swin环境配置
研究背景
检测任务
项⽬代码
Swin-Transformer-Object-Detection code
学习参考(Swin-Transformer源码(已跑通)
环境配置
- readline=8.1=h27cfd23_0
- setuptools=52.0.0=py37h06a4308_0
- six=1.15.0=py37h06a4308_0
- sqlite=3.35.4=hdfb4753_0
- tk=8.6.10=hbc83047_0
- torchvision=0.6.0=py37_cu101
KeyError: "CascadeRCNN: 've_position_bias_table'"
3. 训练启动后关于初始化的 RuntimeError 的问题
问题描述:
RuntimeError: Default process group has not been initialized, please make sure to call init_process_group.
问题原因是⾮分布式训练使⽤了分布式训练的设置
解决⽅式:
runner = dict(type='EpochBasedRunner', max_epochs=36)
## Disable apex
# # runner = dict(type='EpochBasedRunnerAmp', max_epochs=36)
# # do not use mmdet version fp16
# fp16 = None
# optimizer_config = dict(
# type="DistOptimizerHook",
与mmdetection修改⽅式类似。
⾸先是修改配置⽂件configs/swin/cascade_mask_rcnn_swin_ 的类别个数


发布评论