torch.save() torch.load()

在使用pytorch进行深度学习训练的时候，经常会有些内容需要保存下来，保存到硬盘张，不管什么时候我们都可以读取到，那么这个时候，使用torch.save()就可以将内容存储器来，使用torch.load()就可以将存取的内容读取出来。

1.torch.save()使用方法

torch.save()使用的时候需要往里面放进去两个参数，分别是要存储的内容和存储的地址。下面是个例子：

import torch

statemement = {'name':'dalishi','age':19}
torch.save( statemement, 'save_try.pth.tar' )

我要保存的内容是一个字典，那么这个字典就会原原本本的保存下来，这里不只是可以保存字典，其他的任何类型内容都可以保存下来，包括列表、单变量、张量什么的，都没有问题。

2.torch.load()使用方法

对于使用了torch.save()保存了的内容，如何读取出来呢？例子如下：

import torch

save_try = torch.load( 'save_try.pth.tar' )
print(save_try)

运行结果如下：

{'name': 'dalishi', 'age': 19}