平凡之上
博观而约取,厚积而薄发! 守得云开见月明。

#P27-P29

 

完整的模型训练套路

思路步骤:

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

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

            ↓loss计算损失值

            ↓优化器优化模型

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

→每轮结束               ↓for data in test_dataloader

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

 

 

 

#优化代码,正确率

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

().sum  求正确的个数

 

posted on 2021-11-27 16:21  平凡之上  阅读(17)  评论(0编辑  收藏  举报