十分钟从 PyTorch 转 MXNet - 知乎
zhuanlan.zhihu.com › p › 35237659MXNet 跟 PyTorch 的不同主要在下面这几点: 不需要将输入放进 Variable , 但需要将计算放在 mx.autograd.record () 里使得后面可以对其求导 不需要每次梯度清 0,因为新梯度是写进去,而不是累加 step 的时候 MXNet 需要给定批量大小 需要调用 asscalar () 来将多维数组变成标量。 这个样例里 MXNet 比 PyTorch 快两倍。 当然大家对待这样的比较要谨慎。 下一步 更详细的 MXNet 的教程 欢迎给我们留言哪些 PyTorch 的方便之处你希望 MXNet 应该也可以有 发布于 2018-04-02 MXNet PyTorch 文章被以下专栏收录 gluon 让深度学习门槛更低 37 条评论 条评论被折叠 ( 为什么?