11/26日27日学习2

#P27-P29

 

完整的模型训练套路

思路步骤：

准备数据集→DataLoader加载数据集→创建网络模型→创建损失函数→设置优化器→设置训练的一些参数

→开始训练步骤：↓for data in train_dataloader

　　　　　　　　  　　↓loss计算损失值

　　　　　　　　　　  ↓优化器优化模型

→　　　　　　　↓with torch.no_grad():　　　　#取消梯度值

→每轮结束           　　  ↓for data in test_dataloader

   进行测试：　　　　   　　↓loss计算损失值

 

 

 

#优化代码，正确率

 argmax():1.横向比较　　2.纵向比较

().sum　　求正确的个数

 

标签：11,loss,26,27,训练,优化,模型,dataloader,data
来源： https://www.cnblogs.com/upgst/p/15609523.html