搭建神经网络基础综合
以下内容是根据torch官网和莫烦python学习所得
基本步骤
载入数据,训练集,预测集,标注集
搭建网络,即 class Net
实例化网络 net
创建 optimizer
确定损失函数 loss_func
开始训练
计算预测值 predict
计算损失函数值 loss
优化器 zerograd()
损失反馈 loss.backward()
优化器步进 optimizer.step()
import torch
from torch.autograd import V