loading weights by component

[PushpakBhoge512]

Can we load the weight by individual components of a model ?

E.g. say I want to train a mask RCNN but since the augmentations like mosaic etc are not supported here it will be held back

So I decided to train the model in two steps i.e. train RPN separately with RPN configs with all the augmentations. Then use this weight into a mask rcnn config and freeze up to RPN

How can this be achieved?