#P27-P29
完整的模型训练套路
思路步骤:
准备数据集→DataLoader加载数据集→创建网络模型→创建损失函数→设置优化器→设置训练的一些参数
→开始训练步骤:↓for data in train_dataloader
↓loss计算损失值
↓优化器优化模型
→ ↓with torch.no_grad(): #取消梯度值
→每轮结束 ↓for data in test_dataloader
进行测试: ↓loss计算损失值
#优化代码,正确率
argmax():1.横向比较 2.纵向比较
().sum 求正确的个数