PyTorch_神经网络详解

神经网络可使用torch.nn包。nn.Module包含层和方法,forward(input)返回output。 神经网络的典型训练过程以下: ------定义具备一些可学习参数(或权重)的神经网络。 ------迭代输入数据集 ------经过网络处理输入 ------计算损失(输出离正确有多远) ------将梯度传播回网络参数 ------更新网络的权重,一般使用简单的更新规则:weight
相关文章
相关标签/搜索