8 读取和存储Tensor

前程似锦蝈蝈

于 2021-08-18 18:03:27 发布

阅读量350

点赞数

分类专栏：暑假每日记

本文链接：https://blog.youkuaiyun.com/qq_45654059/article/details/119786088

版权

关键词由优快云通过智能技术生成

15 篇文章

订阅专栏

这篇博客详细介绍了如何使用PyTorch进行模型的保存和加载。包括基本的张量存储，列表和字典类型的数据保存，以及模型状态字典的保存与加载方法。这些操作对于模型训练和部署过程中的持久化至关重要。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

torch.save(tensor,'xxx.pt')
l = torch.load('xxx.pt')
# 储存一个tensor列表
torch.save([x,y],'xy.pt')
#储存一个tensor字典
torch.save({'x':x,'y':y},'xy_dict.pt')

torch.save(model.state_dict(),PATH)# 后缀为.pt
model = TheModelClass(*args,**kwargs)
model.load_state_dict(torch.load(PATH))