当我们花费大量的精力训练完网络,下次预测数据时不想再训练一次时,这时候torch.save(),torch.load()就要登场了。

基于pytorch的保存和加载模型参数的方法