笔记 - PyTorch学习

[zengbin93]

PyTorch is a deep learning framework for fast, flexible experimentation. Tensors and Dynamic neural networks in Python with strong GPU acceleration.

• 官网： https://pytorch.org/
• github: https://github.com/pytorch/pytorch
• 官方文档： https://pytorch.org/docs/stable/index.html
• examples: https://github.com/pytorch/examples/
• tutorials: https://pytorch.org/tutorials/
• 60分钟入门教程： https://pytorch.org/tutorials/beginner/deep_learning_60min_blitz.html
• torchtext: http://torchtext.readthedocs.io/en/latest/
• torchvision: https://pytorch.org/docs/master/torchvision/
• 中文文档：https://pytorch-cn.readthedocs.io/zh/latest/

[zengbin93]

模型的保存与恢复

第一种：只保存和加载模型参数：

# 保存
torch.save(the_model.state_dict(), PATH)
# 加载
the_model = TheModelClass(*args, **kwargs)
the_model.load_state_dict(torch.load(PATH))

第二种：保存和加载整个模型

# 保存
torch.save(the_model, PATH)
# 加载
the_model = torch.load(PATH)

在这种情况下，序列化的数据被绑定到特定的类和固定的目录结构，所以当在其他项目中使用时，或者在一些严重的重构器之后它可能会以各种方式break。

[zengbin93]

Tensor和Variable

Tensor和Variable是pytorch中的两种数据结构。 Variable实际是一个能够记录操作信息并且自动求导的容器，Variable数据类型可以进行反向传播来进行自动求导。

详细比较，参考：https://ptorch.com/docs/4/pytorch-video-variable