跟着例子学习 PyTorch

译者:@yongjay13@speedmancs

校对者:@bringtree

Author: Justin Johnson

这个教程通过一些单独的示例介绍了 PyTorch 的基本概念.

PyTorch 的核心部分提供了两个主要功能:

  • 一个类似于 numpy 的n维张量, 但可以在 GPU 上运行
  • 为建立和训练神经网络自动微分

我们将使用全连接的 ReLU 网络作为我们的运行示例. 该网络将有一个隐藏层, 并将使用梯度下降训练去最小化随机数字的预测输出和真实输出之间的欧式距离.

注解:

你可以下载这些单独的例子在页面的底端 <examples-download>.

本章内容目录

  • Tensors
    • Warm-up: numpy
    • PyTorch: Tensors
  • Autograd
    • PyTorch: Variables and autograd
    • PyTorch: Defining new autograd functions
    • TensorFlow: Static Graphs
  • nn module
    • PyTorch: nn
    • PyTorch: optim
    • PyTorch: Custom nn Modules
    • PyTorch: Control Flow + Weight Sharing
  • Examples
    • Tensors
    • Autograd
    • nn module

Tensors

Warm-up: numpy

在介绍 PyTorch 之前, 我们先使用 numpy 实现网络.

Numpy 提供了一个n维的数组对象, 并提供了许多操纵这个数组对象的函数. Numpy 是科学计算的通用框架; Numpy 数组没有计算图, 也没有深度学习, 也没有梯度下降等方法实现的接口. 但是我们仍然可以很容易地使用 numpy 生成随机数据 并将产生的数据传入双层的神经网络, 并使用 numpy 来实现这个网络的正向传播和反向传播:

  1. # -*- coding: utf-8 -*-
  2. import numpy as np
  3. # N 是一个batch的样本数量; D_in是输入维度;
  4. # H 是隐藏层向量的维度; D_out是输出维度.
  5. N, D_in, H, D_out = 64, 1000, 100, 10
  6. # 创建随机的输入输出数据
  7. x = np.random.randn(N, D_in)
  8. y = np.random.randn(N, D_out)
  9. # 随机初始化权重参数
  10. w1 = np.random.randn(D_in, H)
  11. w2 = np.random.randn(H, D_out)
  12. learning_rate = 1e-6
  13. for t in range(500):
  14. # 前向计算, 算出y的预测值
  15. h = x.dot(w1)
  16. h_relu = np.maximum(h, 0)
  17. y_pred = h_relu.dot(w2)
  18. # 计算并打印误差值
  19. loss = np.square(y_pred - y).sum()
  20. print(t, loss)
  21. # 在反向传播中, 计算出误差关于w1和w2的导数
  22. grad_y_pred = 2.0 * (y_pred - y)
  23. grad_w2 = h_relu.T.dot(grad_y_pred)
  24. grad_h_relu = grad_y_pred.dot(w2.T)
  25. grad_h = grad_h_relu.copy()
  26. grad_h[h < 0] = 0
  27. grad_w1 = x.T.dot(grad_h)
  28. # 更新权重
  29. w1 -= learning_rate * grad_w1
  30. w2 -= learning_rate * grad_w2

PyTorch: Tensors

Numpy 是一个伟大的框架, 但它不能利用 GPU 加速它数值计算. 对于现代的深度神经网络, GPU 往往是提供 50倍或更大的加速, 所以不幸的是, numpy 不足以满足现在深度学习的需求.

这里我们介绍一下最基本的 PyTorch 概念: Tensor . PyTorch Tensor 在概念上与 numpy 数组相同: Tensor 是一个n维数组, PyTorch 也提供了很多能在这些 Tensor 上操作的函数. 像 numpy 数组一样, PyTorch Tensor 也和numpy的数组对象一样不了解深度学习,计算图和梯度下降;它们只是科学计算的通用工具.

然而不像 numpy, PyTorch Tensor 可以利用 GPU 加速他们的数字计算. 要在 GPU 上运行 PyTorch 张量, 只需将其转换为新的数据类型.

在这里, 我们将 PyTorch Tensor 生成的随机数据传入双层的神经网络. 就像上面的 numpy 例子一样, 我们需要手动实现网络的正向传播和反向传播:

  1. # -*- coding: utf-8 -*-
  2. import torch
  3. dtype = torch.FloatTensor
  4. # dtype = torch.cuda.FloatTensor # 取消注释以在GPU上运行
  5. # N 批量大小; D_in是输入尺寸;
  6. # H是隐藏尺寸; D_out是输出尺寸.
  7. N, D_in, H, D_out = 64, 1000, 100, 10
  8. # 创建随机输入和输出数据
  9. x = torch.randn(N, D_in).type(dtype)
  10. y = torch.randn(N, D_out).type(dtype)
  11. # 随机初始化权重
  12. w1 = torch.randn(D_in, H).type(dtype)
  13. w2 = torch.randn(H, D_out).type(dtype)
  14. learning_rate = 1e-6
  15. for t in range(500):
  16. # 正向传递:计算预测y
  17. h = x.mm(w1)
  18. h_relu = h.clamp(min=0)
  19. y_pred = h_relu.mm(w2)
  20. # 计算并打印loss
  21. loss = (y_pred - y).pow(2).sum()
  22. print(t, loss)
  23. # 反向传播计算关于损失的w1和w2的梯度
  24. grad_y_pred = 2.0 * (y_pred - y)
  25. grad_w2 = h_relu.t().mm(grad_y_pred)
  26. grad_h_relu = grad_y_pred.mm(w2.t())
  27. grad_h = grad_h_relu.clone()
  28. grad_h[h < 0] = 0
  29. grad_w1 = x.t().mm(grad_h)
  30. # 使用梯度下降更新权重
  31. w1 -= learning_rate * grad_w1
  32. w2 -= learning_rate * grad_w2

Autograd

PyTorch: Variables and autograd

在上面的例子中, 我们不得不手写实现神经网络的正反向传播的代码. 而手写实现反向传播的代码对于一个 小型的双层网络来说是没什么大问题的, 但是在面对大型复杂网络手写方向传播代码就会变得很棘手.

谢天谢地, 我们可以使用 自动微分 来自动化的计算神经网络中的后向传播. PyTorch 中的 autograd 包提供自动微分了这个功能. 使用 autograd 时, 网络的正向传播将定义一个 计算图 ; Tensor 将会成为图中的节点,从输入 Tensor 产生输出 Tensor 的函数将会用图中的( Edge )依赖边表示. 通过计算图来反向传播可以让您轻松计算梯度.

这听起来很复杂, 但是在实践中使用起来相当简单. 我们将 PyTorch 的 Tensor 包装成在 Variable 对象; 一个 Variable 代表一个计算图中的节点. 如果 x 是一个 Variable , 则 x.data 是一个 Tensor , 而 x.grad 是另外一个包含关于 x 的梯度的 Variable .

PyTorch Variable 与 PyTorch Tensor 具有相同的 API: (几乎) 任何您可以在 Tensor 上执行的 操作也适用于 Variable ;该区别在于如果你使用 Variable 定义了一个计算图, Pytorch 允许您自动计算梯度.

这里我们使用 PyTorch 的 Variable 和自动微分来实现我们的双层网络;现在我们不再需要手写任何关于 计算网络反向传播的代码:

  1. # -*- coding: utf-8 -*-
  2. import torch
  3. from torch.autograd import Variable
  4. dtype = torch.FloatTensor
  5. # dtype = torch.cuda.FloatTensor # 取消注释以在GPU上运行
  6. # N 批量大小; D_in是输入尺寸;
  7. # H是隐藏尺寸; D_out是输出尺寸.
  8. N, D_in, H, D_out = 64, 1000, 100, 10
  9. # 创建随机张量来保存输入和输出,并将它们包装在变量中.
  10. # 设置requires_grad = False, 因为在后向传播时, 我们并不需要计算关于这些变量的梯度
  11. x = Variable(torch.randn(N, D_in).type(dtype), requires_grad=False)
  12. y = Variable(torch.randn(N, D_out).type(dtype), requires_grad=False)
  13. # 为权重创建随机张量,并将其包装在变量中.
  14. # 设置requires_grad = True, 因为在后向传播时, 我们需要计算关于这些变量的梯度
  15. w1 = Variable(torch.randn(D_in, H).type(dtype), requires_grad=True)
  16. w2 = Variable(torch.randn(H, D_out).type(dtype), requires_grad=True)
  17. learning_rate = 1e-6
  18. for t in range(500):
  19. # 正向传递:使用变量上的运算来计算预测的y; 这些
  20. # 与我们用于计算使用张量的正向传递完全相同,
  21. # 但我们不需要保留对中间值的引用,
  22. # 因为我们没有实现向后传递.
  23. y_pred = x.mm(w1).clamp(min=0).mm(w2)
  24. # 使用变量上的操作计算和打印损失.
  25. # 现在损失是形状变量 (1,) 并且 loss.data 是形状的张量
  26. # (1,); loss.data[0] 是持有损失的标量值.
  27. loss = (y_pred - y).pow(2).sum()
  28. print(t, loss.data[0])
  29. # 使用autograd来计算反向传递.
  30. # 该调用将使用requires_grad = True来计算相对于所有变量的损失梯度.
  31. # 在这次调用之后 w1.grad 和 w2.grad 将是变量
  32. # 它们分别相对于w1和w2保存损失的梯度.
  33. loss.backward()
  34. # 使用梯度下降更新权重; w1.data 和 w2.data 是张量,
  35. # w1.grad 和 w2.grad 是变量并且 w1.grad.data 和 w2.grad.data
  36. # 是张量.
  37. w1.data -= learning_rate * w1.grad.data
  38. w2.data -= learning_rate * w2.grad.data
  39. # 更新权重后手动将梯度归零
  40. w1.grad.data.zero_()
  41. w2.grad.data.zero_()

PyTorch: Defining new autograd functions

在这层覆盖下, 每个原始的 autograd 操作符实际上是两个函数在张量上运行. 前向传播 函数从输入的 Tensor 计算将要输出的 Tensor . 后向传播 函数接收上一个 Tensor 关于 scalar 的梯度, 以 及计算当前输入 Tensor 对相同 scalar 值的梯度.

在 PyTorch 中, 我们可以通过定义一个 torch.autograd.Function 的子类和 实现 前向传播后向传播 函数来轻松定义自己的 autograd 操作符. 然后我们可以 使用我们新的 autograd 操作符构造一个实例并将其作为一个函数调用, 传递用 Variable 包装了的输入数据的.

在这个例子中我们定义了我们自己的 autograd 函数来执行 ReLU 非线性函数, 并用它来实现我们的双层网络:

  1. # -*- coding: utf-8 -*-
  2. import torch
  3. from torch.autograd import Variable
  4. class MyReLU(torch.autograd.Function):
  5. """
  6. 我们可以通过子类实现我们自己定制的autograd函数
  7. torch.autograd.Function和执行在Tensors上运行的向前和向后通行证.
  8. """
  9. @staticmethod
  10. def forward(ctx, input):
  11. """
  12. 在正向传递中,我们收到一个包含输入和返回张量的张量,其中包含输出.
  13. ctx是一个上下文对象,可用于存储反向计算的信息.
  14. 您可以使用ctx.save_for_backward方法缓存任意对象以用于后向传递.
  15. """
  16. ctx.save_for_backward(input)
  17. return input.clamp(min=0)
  18. @staticmethod
  19. def backward(ctx, grad_output):
  20. """
  21. 在后向传递中,我们收到一个张量,其中包含相对于输出的损失梯度,
  22. 我们需要计算相对于输入的损失梯度.
  23. """
  24. input, = ctx.saved_tensors
  25. grad_input = grad_output.clone()
  26. grad_input[input < 0] = 0
  27. return grad_input
  28. dtype = torch.FloatTensor
  29. # dtype = torch.cuda.FloatTensor # 取消注释以在GPU上运行
  30. # N 批量大小; D_in是输入尺寸;
  31. # H是隐藏尺寸; D_out是输出尺寸.
  32. N, D_in, H, D_out = 64, 1000, 100, 10
  33. # 创建随机张量来保存输入和输出,并将它们包装在变量中.
  34. x = Variable(torch.randn(N, D_in).type(dtype), requires_grad=False)
  35. y = Variable(torch.randn(N, D_out).type(dtype), requires_grad=False)
  36. # 为权重创建随机张量,并将其包装在变量中.
  37. w1 = Variable(torch.randn(D_in, H).type(dtype), requires_grad=True)
  38. w2 = Variable(torch.randn(H, D_out).type(dtype), requires_grad=True)
  39. learning_rate = 1e-6
  40. for t in range(500):
  41. # 为了应用我们的函数,我们使用Function.apply方法.我们把它称为'relu'.
  42. relu = MyReLU.apply
  43. # 正向传递:使用变量上的运算来计算预测的y;
  44. # 我们使用我们的自定义autograd操作来计算ReLU.
  45. y_pred = relu(x.mm(w1)).mm(w2)
  46. # 计算和打印损失
  47. loss = (y_pred - y).pow(2).sum()
  48. print(t, loss.data[0])
  49. # 使用autograd来计算反向传递.
  50. loss.backward()
  51. # 使用梯度下降更新权重
  52. w1.data -= learning_rate * w1.grad.data
  53. w2.data -= learning_rate * w2.grad.data
  54. # 更新权重后手动将梯度归零
  55. w1.grad.data.zero_()
  56. w2.grad.data.zero_()

TensorFlow: Static Graphs

Pytorch 的 autograd 看上去有点像 TensorFlow .两个框架的共同点是他们都是定义了自己的计算图. 和使用自动求微分的方法来计算梯度. 两者之间最大的不同在于 TensorFlow 的计算图是 静态的 和 PyTorch 的计算图是 动态的 .

在 TensorFlow 中, 我们只定义了一次计算图,然后重复执行同一张计算图, 只是输入计算图的数据不同而已. 而在 PyTorch 中, 每个正向传播都会定义一个新的计算图.

静态图很好, 因为您可以预先优化计算图;例如一个框架可能会为了计算效率决定融合一些计算图操作(像:Fused Graph), 或提出 一个多卡或者多机的分布式计算图的策略. 如果您正在重复使用相同的计算图, 那么这个潜在的 昂贵的前期优化可以使用静态图来得以减轻.

一方面来说, 静态图和动态图的控制流是不同的. 对于有些模型我们可能希望对每个数据点执行不同 的计算;例如循环神经网络可能会被展开为对每个数据的不同的长度的时间步数;这个展开可以用循 环来实现. 循环结构的静态图需要成为计算图的一部分;为此 TensorFlow 提供 tf.scan 操作符 用于将重复的结构嵌入到计算图中. 而动态计算图的情况比较简单: 因为我们设计的计算图可以对每个不同长度的输入随机应变. 我们可以使用正常的命令式代码对每个不同长度的输入执行计算.

为了与上面的 PyTorch autograd 例子进行对比, 我们在这里也使用 TensorFlow 创建简单的两层神经网络:

  1. # -*- coding: utf-8 -*-
  2. import tensorflow as tf
  3. import numpy as np
  4. # 首先我们设置计算图:
  5. # N 批量大小; D_in是输入尺寸;
  6. # H是隐藏尺寸; D_out是输出尺寸.
  7. N, D_in, H, D_out = 64, 1000, 100, 10
  8. # 为输入数据和目标数据创建占位符;
  9. # 当我们执行图时,这些将被填充真实的数据.
  10. x = tf.placeholder(tf.float32, shape=(None, D_in))
  11. y = tf.placeholder(tf.float32, shape=(None, D_out))
  12. # 为权重创建变量并用随机数据初始化它们.
  13. # 一个TensorFlow变量在图的执行中保持其值.
  14. w1 = tf.Variable(tf.random_normal((D_in, H)))
  15. w2 = tf.Variable(tf.random_normal((H, D_out)))
  16. # 正向传递:使用TensorFlow Tensors上的运算来计算预测的y.
  17. # 请注意此代码实际上并未执行任何数字操作;
  18. # 它只是设置我们稍后将执行的计算图.
  19. h = tf.matmul(x, w1)
  20. h_relu = tf.maximum(h, tf.zeros(1))
  21. y_pred = tf.matmul(h_relu, w2)
  22. # 使用TensorFlow张量上的操作计算损失
  23. loss = tf.reduce_sum((y - y_pred) ** 2.0)
  24. # 计算相对于w1和w2的损失梯度.
  25. grad_w1, grad_w2 = tf.gradients(loss, [w1, w2])
  26. # 使用梯度下降更新权重.
  27. # 要实际更新权重,我们需要在执行图时评估new_w1和new_w2.
  28. # 请注意,在TensorFlow中,更新权值的行为是计算图的一部分
  29. # 在PyTorch中,这发生在计算图之外.
  30. learning_rate = 1e-6
  31. new_w1 = w1.assign(w1 - learning_rate * grad_w1)
  32. new_w2 = w2.assign(w2 - learning_rate * grad_w2)
  33. # 现在我们已经构建了计算图,所以我们输入一个TensorFlow会话来实际执行图.
  34. with tf.Session() as sess:
  35. # 运行一次图形初始化变量w1和w2.
  36. sess.run(tf.global_variables_initializer())
  37. # 创建包含输入x和目标y的实际数据的numpy数组
  38. x_value = np.random.randn(N, D_in)
  39. y_value = np.random.randn(N, D_out)
  40. for _ in range(500):
  41. # 多次执行图. 每次执行时,
  42. # 我们都想将x_value绑定到x,将y_value绑定到y,用feed_dict参数指定.
  43. # 每次我们执行图时,我们都想计算损失值new_w1 和 new_w2;
  44. # 这些张量的值作为numpy数组返回.
  45. loss_value, _, _ = sess.run([loss, new_w1, new_w2],
  46. feed_dict={x: x_value, y: y_value})
  47. print(loss_value)

nn module

PyTorch: nn

计算图( Computational graphs )和 autograd 是一个非常强大的定义复杂的运算符并自动地导出的范式;然而对于 大型的神经网络, 原始的 autograd 仍然显得有点太低级.

当我们创建神经网络时, 我们经常思考如何设计安排 layer , 以及一些在训练过程中网络会学习到的 learnable parameters

在TensorFlow中, 像 Keras, TensorFlow-Slim, 和 TFLearn 通过构建对神经网络有用的原始计算图提供更高层次的抽象.

在 PyTorch 中, nn 包起了同样的作用. nn 包定义了一组 Modules , 大致相当于神经网络层. 模块接收输入变量并进行计算输出变量, 但也可以保持内部状态, 如 用 Variable 包装的 learnable parameters . nn 包 也定义了一系列在训练神经网络时比较常用的损失函数.

在这个例子中, 我们使用 nn 包来实现我们的双层神经网络:

  1. # -*- coding: utf-8 -*-
  2. import torch
  3. from torch.autograd import Variable
  4. # N 批量大小; D_in是输入尺寸;
  5. # H是隐藏尺寸; D_out是输出尺寸.
  6. N, D_in, H, D_out = 64, 1000, 100, 10
  7. # 创建随机张量来保存输入和输出,并将它们包装在变量中.
  8. x = Variable(torch.randn(N, D_in))
  9. y = Variable(torch.randn(N, D_out), requires_grad=False)
  10. # 使用nn包将我们的模型定义为一系列图层.
  11. # nn.Sequential是一个包含其他模块的模块,并将它们按顺序应用以产生其输出.
  12. # 每个线性模块使用线性函数计算来自输入的输出,并保存内部变量的权重和偏差.
  13. model = torch.nn.Sequential(
  14. torch.nn.Linear(D_in, H),
  15. torch.nn.ReLU(),
  16. torch.nn.Linear(H, D_out),
  17. )
  18. # nn包还包含流行损失函数的定义;
  19. # 在这种情况下,我们将使用均方差(MSE)作为我们的损失函数.
  20. loss_fn = torch.nn.MSELoss(size_average=False)
  21. learning_rate = 1e-4
  22. for t in range(500):
  23. # 正向传递:通过将x传递给模型来计算预测的y.
  24. # 模块对象会覆盖__call__运算符,因此您可以将它们称为函数.
  25. # 这样做时,您将输入数据的变量传递给模块,并生成输出数据的变量.
  26. y_pred = model(x)
  27. # 计算和打印损失.
  28. # 我们传递包含y的预测值和真值的变量,并且损失函数返回包含损失的变量.
  29. loss = loss_fn(y_pred, y)
  30. print(t, loss.data[0])
  31. # 在运行反向传递之前将梯度归零.
  32. model.zero_grad()
  33. # 向后传递:计算相对于模型的所有可学习参数的损失梯度.
  34. # 在内部,每个模块的参数都存储在变量require_grad = True中,
  35. # 因此该调用将计算模型中所有可学习参数的梯度.
  36. loss.backward()
  37. # 使用梯度下降更新权重.
  38. # 每个参数都是一个变量,所以我们可以像我们以前那样访问它的数据和梯度.
  39. for param in model.parameters():
  40. param.data -= learning_rate * param.grad.data

PyTorch: optim

到目前为止, 我们一直通过手动更新的方法更新模型的可学习参数( learnable parameters )的权重 .data 这对于简单的优化算法像随机梯度下降来还算轻松, 但是在实际中我们经常使用更巧妙的 优化器来训练神经网络, 如 AdaGrad, RMSProp, Adam 等.

PyTorch 中的 optim 包包含了一些优化器的算法, 并提供了一些常用优化器的使用.

在这个例子中, 虽然我们将像之前一样使用 nn 包来定义我们的模型, 但是我们这次将使用由 optim 包提供的Adam算法来更新模型:

  1. # -*- coding: utf-8 -*-
  2. import torch
  3. from torch.autograd import Variable
  4. # N 批量大小; D_in是输入尺寸;
  5. # H是隐藏尺寸; D_out是输出尺寸.
  6. N, D_in, H, D_out = 64, 1000, 100, 10
  7. # 创建随机张量来保存输入和输出,并将它们包装在变量中.
  8. x = Variable(torch.randn(N, D_in))
  9. y = Variable(torch.randn(N, D_out), requires_grad=False)
  10. # 使用nn包来定义我们的模型和损失函数.
  11. model = torch.nn.Sequential(
  12. torch.nn.Linear(D_in, H),
  13. torch.nn.ReLU(),
  14. torch.nn.Linear(H, D_out),
  15. )
  16. loss_fn = torch.nn.MSELoss(size_average=False)
  17. # 使用优化包来定义一个优化器,它将为我们更新模型的权重.
  18. # 在这里,我们将使用 Adam;这个 optim 包包含许多其他优化算法.
  19. # Adam构造函数的第一个参数告诉优化器应该更新哪个Variables.
  20. learning_rate = 1e-4
  21. optimizer = torch.optim.Adam(model.parameters(), lr=learning_rate)
  22. for t in range(500):
  23. # 正向传递:通过将x传递给模型来计算预测的y.
  24. y_pred = model(x)
  25. # 计算和打印损失函数.
  26. loss = loss_fn(y_pred, y)
  27. print(t, loss.data[0])
  28. # 在向后传递之前,使用优化器对象为其要更新的变量(这是模型的可学习权重)的所有梯度归零.
  29. # 这是因为默认情况下,只要调用.backward(),梯度就会在缓冲区中累积(即不会被覆盖).
  30. # 查看torch.autograd.backward的文档以获取更多详细信息.
  31. optimizer.zero_grad()
  32. # 向后传递:计算损失函数相对于模型参数的梯度
  33. loss.backward()
  34. # 在优化器上调用step函数会更新其参数
  35. optimizer.step()

PyTorch: Custom nn Modules

有时你会想要使用比现有模块组合更复杂的特殊模型;对于这些情况, 你可以 通过继承 nn.Module 来定义你自己的模块, 并定义一个 forward 来实现模块接收输入 Variable 并使用其他模块输出的 Variable 和 其他 autograd 操作.

在这个例子中, 我们使用了我们之前已经实现的双层网络来作为一个自定义的模块子类:

  1. # -*- coding: utf-8 -*-
  2. import torch
  3. from torch.autograd import Variable
  4. class TwoLayerNet(torch.nn.Module):
  5. def __init__(self, D_in, H, D_out):
  6. """
  7. 在构造函数中,我们实例化两个nn.Linear模块并将它们分配为成员变量.
  8. """
  9. super(TwoLayerNet, self).__init__()
  10. self.linear1 = torch.nn.Linear(D_in, H)
  11. self.linear2 = torch.nn.Linear(H, D_out)
  12. def forward(self, x):
  13. """
  14. 在forward函数中,我们接受一个变量的输入数据,我们必须返回一个变量的输出数据.
  15. 我们可以使用构造函数中定义的模块以及变量上的任意运算符.
  16. """
  17. h_relu = self.linear1(x).clamp(min=0)
  18. y_pred = self.linear2(h_relu)
  19. return y_pred
  20. # N 批量大小; D_in是输入尺寸;
  21. # H是隐藏尺寸; D_out是输出尺寸.
  22. N, D_in, H, D_out = 64, 1000, 100, 10
  23. # 创建随机张量来保存输入和输出,并将它们包装在变量中.
  24. x = Variable(torch.randn(N, D_in))
  25. y = Variable(torch.randn(N, D_out), requires_grad=False)
  26. # 通过实例化上面定义的类来构建我们的模型
  27. model = TwoLayerNet(D_in, H, D_out)
  28. # 构建我们的损失函数和优化器.
  29. # 对SGD构造函数中的model.parameters()的调用将包含作为模型成员的两个nn.Linear模块的可学习参数.
  30. criterion = torch.nn.MSELoss(size_average=False)
  31. optimizer = torch.optim.SGD(model.parameters(), lr=1e-4)
  32. for t in range(500):
  33. # 正向传递:通过将x传递给模型来计算预测的y
  34. y_pred = model(x)
  35. # 计算和打印损失
  36. loss = criterion(y_pred, y)
  37. print(t, loss.data[0])
  38. # 梯度置零, 执行反向传递并更新权重.
  39. optimizer.zero_grad()
  40. loss.backward()
  41. optimizer.step()

PyTorch: Control Flow + Weight Sharing

作为一个动态图和权值共享的例子, 我们实现一个奇葩的模型: 随机1-4次重复搭建同个正向传播的全连接 的 ReLU 网络, 并且多个隐藏层使用相同的权重来计算最内层隐藏层(译者注: 这里的相同权重,是指随机1-4次重复搭建的这个middle_linear).

对于这个模型, 我们可以使用普通的 Python 流程控制语句来实现循环, 而且我们可以在定义前向传 播时通过简单地重复使用相同的模块实现 middle_linear 层的权重共享.

我们可以很容易地将这个模型作为 Module 子类来实现:

  1. # -*- coding: utf-8 -*-
  2. import random
  3. import torch
  4. from torch.autograd import Variable
  5. class DynamicNet(torch.nn.Module):
  6. def __init__(self, D_in, H, D_out):
  7. """
  8. 在构造函数中,我们构造了三个nn.Linear实例,我们将在正向传递中使用它们.
  9. """
  10. super(DynamicNet, self).__init__()
  11. self.input_linear = torch.nn.Linear(D_in, H)
  12. self.middle_linear = torch.nn.Linear(H, H)
  13. self.output_linear = torch.nn.Linear(H, D_out)
  14. def forward(self, x):
  15. """
  16. 对于模型的正向通道,我们随机选择0,1,2或3,
  17. 并重复使用多次计算隐藏层表示的middle_linear模块.
  18. 由于每个正向通道都会生成一个动态计算图,因此在定义模型的正向通道时,
  19. 我们可以使用普通的Python控制流操作符(如循环或条件语句).
  20. 在这里我们也看到,定义计算图时多次重复使用相同模块是完全安全的.
  21. 这是Lua Torch的一大改进,每个模块只能使用一次.
  22. """
  23. h_relu = self.input_linear(x).clamp(min=0)
  24. for _ in range(random.randint(0, 3)):
  25. h_relu = self.middle_linear(h_relu).clamp(min=0)
  26. y_pred = self.output_linear(h_relu)
  27. return y_pred
  28. # N 批量大小; D_in是输入尺寸;
  29. # H是隐藏尺寸; D_out是输出尺寸.
  30. N, D_in, H, D_out = 64, 1000, 100, 10
  31. # 创建随机张量来保存输入和输出,并将它们包装在变量中.
  32. x = Variable(torch.randn(N, D_in))
  33. y = Variable(torch.randn(N, D_out), requires_grad=False)
  34. # 通过实例化上面定义的类来构建我们的模型
  35. model = DynamicNet(D_in, H, D_out)
  36. # 构建我们的损失函数和优化器.
  37. # 用随机梯度下降训练这个奇怪的模型非常困难,所以我们使用动量
  38. criterion = torch.nn.MSELoss(size_average=False)
  39. optimizer = torch.optim.SGD(model.parameters(), lr=1e-4, momentum=0.9)
  40. for t in range(500):
  41. # 正向传递:通过将x传递给模型来计算预测的y
  42. y_pred = model(x)
  43. # 计算和打印损失
  44. loss = criterion(y_pred, y)
  45. print(t, loss.data[0])
  46. # 零梯度执行反向传递并更新权重.
  47. optimizer.zero_grad()
  48. loss.backward()
  49. optimizer.step()

Examples

你可以在这里浏览上网提到的例子

Tensors

http://pytorch.apachecn.org/cn/tutorials/_images/sphx_glr_two_layer_net_numpy_thumb.png

Warm-up: numpy

http://pytorch.apachecn.org/cn/tutorials/_images/sphx_glr_two_layer_net_tensor_thumb.png

PyTorch: Tensors

Autograd

http://pytorch.apachecn.org/cn/tutorials/_images/sphx_glr_two_layer_net_autograd_thumb.png

PyTorch: 变量和autograd

http://pytorch.apachecn.org/cn/tutorials/_images/sphx_glr_two_layer_net_customfunctionthumb.png

PyTorch: 定义新的autograd函数

http://pytorch.apachecn.org/cn/tutorials/_images/sphx_glr_tf_two_layer_net_thumb.png

TensorFlow: 静态图

nn module

http://pytorch.apachecn.org/cn/tutorials/_images/sphx_glr_two_layer_net_nn_thumb.png

PyTorch: nn包

http://pytorch.apachecn.org/cn/tutorials/_images/sphx_glr_two_layer_net_optim_thumb.png

PyTorch: optim包

http://pytorch.apachecn.org/cn/tutorials/_images/sphx_glr_two_layer_net_module_thumb.png

PyTorch: 定制化nn模块

http://pytorch.apachecn.org/cn/tutorials/_images/sphx_glr_dynamic_net_thumb.png

PyTorch: 动态控制流程 + 权重共享