可以使用VOC数据集做SWA训练吗

[TTXXXL]

使用VOC数据集训练验证阶段会报错

[hyz-xmaster]

Hi, you can make the following changes to enable the SWA training on the VOC dataset:

changing save_best='bbox_mAP' to save_best='auto', rule='greater' at this line and this line

changing swa_load_from = 'best_bbox_mAP.pth' to swa_load_from = 'best_AP50.pth'

changing if cfg.swa_load_from == 'best_bbox_mAP.pth' to if cfg.swa_load_from == 'best_bbox_mAP.pth' or cfg.swa_load_from == 'best_AP50.pth'

[TTXXXL]

thank you