摘要: 文章目录 Neural Networks定义网络损失函数反向传播更新权重 Neural Networks 使用torch.nn包来构建神经网络。 上一讲已经讲过了autograd,nn包依赖autograd包来定义模型并求导。 一个nn.Module包含各个层和一个forward(input)方法, 阅读全文
posted @ 2021-01-26 11:37 野哥李 阅读(8) 评论(0) 推荐(0) 编辑
摘要: 文章目录 Autograd: 自动求导机制张量(Tensor)梯度 Autograd: 自动求导机制 PyTorch 中所有神经网络的核心是 autograd 包。 我们先简单介绍一下这个包,然后训练第一个简单的神经网络。 autograd包为张量上的所有操作提供了自动求导。 它是一个在运行时定义的 阅读全文
posted @ 2021-01-26 11:08 野哥李 阅读(8) 评论(0) 推荐(0) 编辑