1、激活pytorch环境,并打开jupyter notebook编辑环境

2、第一步,导入训练神经网络的相关torch库的模块。

3、第二步,对超参数进行初始化

4、第三步,导入训练数据,训练数据会被整合成torch.tensor的形式。

5、第四步,设计构建神经网络结构,并到导入该网络结构,然后确定优化器和损失函数。

6、第五步,对神经网络进行分批训练。

7、最后,导入matplotlib模块,显示神经网络的训练损失。

时间:2024-10-20 01:28:41
1、激活pytorch环境,并打开jupyter notebook编辑环境
2、第一步,导入训练神经网络的相关torch库的模块。
3、第二步,对超参数进行初始化
4、第三步,导入训练数据,训练数据会被整合成torch.tensor的形式。
5、第四步,设计构建神经网络结构,并到导入该网络结构,然后确定优化器和损失函数。
6、第五步,对神经网络进行分批训练。
7、最后,导入matplotlib模块,显示神经网络的训练损失。