DAY33
简单的神经网络
-
PyTorch和cuda的安装
-
查看显卡信息的命令行命令(cmd中使用)
-
cuda的检查
-
简单神经网络的流程
1.数据预处理(归一化、转换成张量)
2.模型的定义
1.继承nn.Module类
2.定义每一个层
3.定义前向传播流程
3.定义损失函数和优化器
4. 定义训练流程
5.可视化loss过程
作业:今日的代码,要做到能够手敲。这已经是最简单最基础的版本了。
@浙大疏锦行