RuntimeError: Error(s) in loading state_dict for Sequential about pytorch-ssd

[qfgaohao]

@abcp4 plz try to replace '--base_net' with '--pretrained_ssd' in your command, as you are using a pretrained ssd to initialize the model. '--base_net' expects a pretained base net model, as it only initialises the base net. '--pretrained_ssd' uses pretrained weights to initialise all layers except the prediction head.

[abcp4]

It worked now!! Thank you very much!

[zhanglaplace]

@qfgaohao using --pretrained_ssd finetuning model ,the loss is not match the pretrain model.