文章目录 前言 代码实现 1、导入数据库 关于torch.manual_seed(1) 2、参数设置 3、数据导入 4、定义训练函数 4.1 nn.CrossEntropyLoss() 4.2 .detach() 4.3 .size VS .shape 4.4 .to(DEVICE) 4.5 .max() 4.6 optimizer.zero_grad() 4.7 len(data
