swin transformer训练自己的数据集报错,出现IndexError: tuple index out of range
考虑是不是需要停用apex
具体措施:
在configs/swin/mask_rcnn_swin_tiny_patch4_window7_mstrain_480-800_adamw_3x_coco.py
中:
1、
runner = dict(type='EpochBasedRunnerAmp', max_epochs=36) 改为 runner = dict(type='EpochBasedRunner', max_epochs=36)
2、注释以下字段停用apex:
fp16 = None
optimizer_config = dict(
type="DistOptimizerHook",
update_interval=1,
grad_clip=None,
coalesce=True,
bucket_size_mb=-1,
use_fp16=True,
)