深度学习——模型的加载与保存

## 保存way1

torch.save(net,'net.pth')    # 保存了模型与参数

## 对应加载way1

model = torch.load('net.pth')    # 模型路径

## 保存way2

torch.save(net.state_ddict(), 'net.pth')    # 只保存参数，字典型

## 对应加载way2

net=model()
net.load_state_dict(torch.load('net.pth'))  # 模型路径