mmrotate icon indicating copy to clipboard operation
mmrotate copied to clipboard

backbone 预训练权重加载

Open zhu011 opened this issue 2 years ago • 2 comments

如果对backbone网络改动了一些,在加载原来预训练权重时,指令还可以是这个init_cfg=dict(type='Pretrained', checkpoint='torchvision://resnet50'))吗?如果需要改动,需要在哪里去改

zhu011 avatar Oct 13 '22 11:10 zhu011

We recommend using English or English & Chinese for issues so that we could have broader discussion.

mm-assistant[bot] avatar Oct 13 '22 11:10 mm-assistant[bot]

Yes, you can continue to use ResNet50 as your pre-training model. Because you have made changes to the backbone, some of the original parameters will not match. Usually, changes to the backbone network require us to use ImageNet or COCO dataset to train a model from scratch.

zytx121 avatar Oct 19 '22 01:10 zytx121