如果您在训练时使用了 DataParallel,则加载的权重文件可能包含额外的模块前缀(如 module)。您需要处理这些前缀以匹配您的模型。
# 加载模型权重文件
weights = torch.load('model_weights.pth')
# 如果您在训练时使用了 DataParallel,则需要处理额外的模块前缀
if 'module' in list(weights.keys())[0]:
weights = {k.replace('module.', ''): v for k, v in weights.items()