pytorch模型加载报错：Error(s) in loading state_dict for PGAT_Model:size mismatch for X

问题描述

pytorch模型加载：
model.load_state_dict(torch.load(checkpoint_file) )
报错：Error(s) in loading state_dict for PGAT_Model:size mismatch for X

解决方法

1、检查模型的超参数是否和训练时的一致。batch size，hidden_dim, layer_size等。
2、如果原机器上能正确运行，迁移到新机器上报错，检查路径问题。使用os.path.exists()判断文件是否存在。建议使用相对路径，这样即使迁移到新环境也毋需担心路径问题。
3、最后还没解决，可能是PyTorch版本环境不一致、torch.nn.DataParallel()关键字不匹配等，用以下代码加载模型：

model.load_state_dict(torch.load(checkpoint_file)，False)