NothingToSay99

> Please @wh2333 I'm facing a problem when i load the pretrained model , specially when it reads the checkpoint this is the error .. how did you kindly solve...

Hi~ I meet the same problem with you. Have you solved it？ and please share the method if you could

RuntimeError: Error(s) in loading state_dict for DenseASPP: Unexpected key(s) in state_dict: "features.denseblock1.denselayer1.norm.1.weight", "features.denseblock1.denselayer1.norm.1.bias", "features.denseblock1.denselayer1.norm.1.running_mean", "features.denseblock1.denselayer1.norm.1.running_var", "features.denseblock1.denselayer1.conv.1.weight", "features.denseblock1.denselayer1.norm.2.weight", "features.denseblock1.denselayer1.norm.2.bias", "features.denseblock1.denselayer1.norm.2.running_mean", "features.denseblock1.denselayer1.norm.2.running_var", "features.denseblock1.denselayer1.conv.2.weight", "features.denseblock1.denselayer2.norm.1.weight", "features.denseblock1.denselayer2.norm.1.bias", "features.denseblock1.denselayer2.norm.1.running_mean", "features.denseblock1.denselayer2.norm.1.running_var", "features.denseblock1.denselayer2.conv.1.weight", "features.denseblock1.denselayer2.norm.2.weight", "features.denseblock1.denselayer2.norm.2.bias", "features.denseblock1.denselayer2.norm.2.running_mean", "features.denseblock1.denselayer2.norm.2.running_var",...

I met the same issue! With the same env, using nvidia A100 create the model, then loading it on nvidia 3090, the error ‘’RuntimeError: [Error thrown at core/runtime/TRTEngine.cpp:42] Expected most_compatible_device...

> Thanks for your reply, looking forward to your work!