基于 MNIST 数据集,先手动实现一个完整的多分类神经网络(前向/反向/参数更新),再用 PyTorch 的 nn.Module/CrossEntropyLoss/SGD 重写,对比两者理解框架帮我们做了什么
标签:nn.Module
共 1 篇相关内容,按发布时间排列。
共 1 篇相关内容,按发布时间排列。
基于 MNIST 数据集,先手动实现一个完整的多分类神经网络(前向/反向/参数更新),再用 PyTorch 的 nn.Module/CrossEntropyLoss/SGD 重写,对比两者理解框架帮我们做了什么