model = ConvNeXtV2(depths=[3, 3, 27, 3], dims=[128, 256, 512, 1024]) params=model.state_dict() #获得模型的原始状态以及参数。 for k,v in params.items(): print(k) #只打印key值,不打印具体参数。
pth文件网络,结构可视化_pth文件可视化_我来保fu小仙女的博客-CSDN博客
两组程序分别运行查看不相同处
然后根据
params = {k.replace('module.',''):v for k,v in checkpoint['state_dict'].items()} #替换将要载入的参数的键的不匹配部分
进行替换
更快解决方式
torch.load('****.pth',, strict=False)