question about load checkpoint about imted HOT 4 CLOSED

Hi，@menglongyue，it's normal in the training process. We load the pre-trained weights for encoder and decoder seperately. The unexpected keys in the log are the parameters of the encoder when loading weights for the decoder(MAEBoxHead).

from imted.

@LiewFeng Thanks for you reply very much, i understand it now. another place i do not understand, how to devide foreground and background, could you please explain it?

from imted.

@menglongyue For the base training of COCO in the few-shot setting, there are only 60 foreground classes and 1 background class, whhile for the finetuing, there are 80 foreground classes and 1 background class. Thus, when loading weights from the checkpoint of base training, we need to move the class weight of background to the last.

from imted.

@LiewFeng ok, i get it. Thank you very much!

from imted.