Pytorch网络-3

定义网络 建一个class 定义网络的层 定义网络的前向传播，反向传播会自动推导。 Dataloader 2. 优化器&损失函数 训练过程 model.train() 指明这个网络有梯度，要更新参数。 optimizer.zero_grad() 优化器清零。 output = model(data) 计算前传得到预测的输出。 loss=F.nall_loss(output, target) 计算损

>>阅读原文<<