torch.load - torch.load技术,学习,经验文章

奉系坤阀

1 年前

Pytorch读写张量文件目录一、加载和保存张量1、直接读写张量2、读写张量列表3、读写张量字典二、加载和保存模型参数对于单个张量，我们可以直接调用`load`和`save`函数分别读写它们。这两个函数都要求我们提供一个名称，`save`要求将要保存的变量作为输入。