【pytorch】如何在pytorch自定义优化函数?
pytorch 可以说是最简单易学的深度学习库了,为了加快神经网络的学习效率我们通常会使用一些加速技巧,但很多人都是使用的torch.optim自带的优化器,本文就将介绍如何自定义一个优化器并比较其与主流算法的加速效果1.准备工作首先下载minst数据库作为测试数据集合minst 我们需要train-images.idx3-ubyte和train-labels.idx1-ub...
原创
2018-08-17 11:06:43 ·
8088 阅读 ·
4 评论