KeyError: 'MaskRCNN: "pvt_v2_b1: \'pretrained\'"'

whai362 / PVT

Official implementation of PVT series

Apache License 2.0

1.74k stars 246 forks source link

KeyError: 'MaskRCNN: "pvt_v2_b1: \'pretrained\'"' #105

Closed yuhua666 closed 1 year ago

yuhua666 commented 1 year ago

I can't load the pretrained model. when I run "CUDA_VISIBLE_DEVICES=0,1 PORT=29501 ./dist_train.sh configs/mask_rcnn_pvt_v2_b0 _fpn_1x_coco.py 2 --cfg-options model.pretrained=pretrain/pvt_v2_b0.pth", the error occurs. The other config is the same situation.