问题：加载训练模型出错
解决方法：查看自己保存模型的方法，然后修改相应的加载方法
1：state_dict方式
保存：torch.save(model.state_dict(), 路径) # 推荐的文件后缀名是pt或pth
加载：model = TheModelClass(*args, **kwargs)
model.load_state_dict(torch.load(路径))
2: 直接保存方式
保存：torch.save(model, 路径)
加载：model = torch.load(路径)
