PyTorch 学习笔记(六):PyTorch hook 和关于 PyTorch backward 过程的理解

PyTorch 学习笔记(六):PyTorch hook 和关于 PyTorch backward 过程的理解

在看pytorch官方文档的时候,发现在nn.Module部分和Variable部分均有hook的身影。感到很…

PyTorch 学习笔记(五):存储和恢复模型并查看参数

PyTorch 学习笔记(五):存储和恢复模型并查看参数

在模型完成训练后,我们需要将训练好的模型保存为一个文件供测试使用,或者因为一些原因我们需要继续之前的状态训练之…

PyTorch 学习笔记(四):自定义 Dataset 和输入流

PyTorch 学习笔记(四):自定义 Dataset 和输入流

什么是Datasets: 在输入流水线中,我们看到准备数据的代码是这么写的data = datasets.CI…

PyTorch 学习笔记(三):自动求导

PyTorch 学习笔记(三):自动求导

Backward过程中排除子图 pytorch的BP过程是由一个函数决定的,loss.backward(), …

PyTorch 学习笔记(二):关于Gradient

PyTorch 学习笔记(二):关于Gradient

在BP的时候,pytorch是将Variable的梯度放在Variable对象中的,我们随时都可以使用Vari…

PyTorch 学习笔记(一):什么是 PyTorch

PyTorch 学习笔记(一):什么是 PyTorch

PyTorch是一个动态的建图的工具。不像Tensorflow那样,先建图,然后通过feed和run重复执行建…

返回顶部