pytorch实现多层网络

引入模块,读取数据
构建计算图(构建网络模型)
损失函数与优化器
开始训练模型
对训练的模型预测结果进行评估
发现了一个写的超级好的博客:https://www.cnblogs.com/pprp/p/9748901.html
在这里插入图片描述
在这里插入图片描述
对于构建神经网络,主要分以下几个步骤
1.通过写一个类,来构造你的模型
2.构建loss和优化器
3.开始训练 Forward -> compute loss -> backward -> update
Forward: y_pred = model(x_data)
Compute loss: loss = criterion(y_pred,y_data)
Backward: optimizer.zero_grad() && loss.backward()
Update: optimizer.step()

猜你喜欢

转载自blog.csdn.net/qq_42017043/article/details/89203556