(二)PyTorch

61 阅读1分钟
  • PyTorch:是用python写的一个机器学习框架,好处是,把高维的矩阵运算用GPU去做加速运算,另外,把梯度运算打包的非常漂亮。
  • Tensor:高维阵列

结合前面深度学习中了解到的神经网络训练的整个过程,包括训练、验证、测试。将这整个过程使用PyTorch实现如下:

image.png

1. 资料加载

image.png

2. 定义模型

image.png

3. 定义Loss模型

image.png

4. 模型优化

image.png

完整的过程如下:

image.png

image.png

image.png

image.png