How can I get small_pvt.pth?

[SamMohel]

I run your main.py .. I'm confusing what this class do ? it gave me the accuracy for 500 epoch and loss of them right ? and when I tried to train my images by this command
'dist_train.sh configs/retinanet_pvt_s_fpn_1x_coco_640.py 1'

I got that small_pvt.pth not found .. excuse me does that will be the weights ? Or checkpoints ?

Does small_pvt.pth here
https://drive.google.com/file/d/1vtcyoU8KUqNzktlMGXZrYcMRsNNiVZFQ/view?usp=sharing
For imagenet ? But how can I got pth file.if the dataset.is different . Appreciating your reply. Thanks

[whai362]

Hi, pvt_small.pth is the pre-trained weights of the backbone, not the checkpoint of retinanet_pvt_s_fpn_1x_coco_640.py.

You could create the folder pretrained/, and put the pvt_small.pth download from https://drive.google.com/file/d/1vtcyoU8KUqNzktlMGXZrYcMRsNNiVZFQ/view?usp=sharing in the folder.

[SamMohel]

Thanks so this weights for any datasets ?

[whai362]

Thanks so this weights for any datasets ?

Yes.