美文网首页
gluon.Trainer()

gluon.Trainer()

作者: elementmaster | 来源:发表于2020-03-13 14:44 被阅读0次

#定义优化算法

#Trainer实例,学习率=0.03小批量随机梯度下降(sgd)为优化算法

#迭代net实例

#通过collect_params获取

from mxnet import gluon

trainer=gluon.Trainer(net.collect_params(),'sgd',{'learning_rate':0.03})

相关文章

  • gluon.Trainer()

    #定义优化算法 #Trainer实例,学习率=0.03小批量随机梯度下降(sgd)为优化算法 #迭代net实例 #...

网友评论

      本文标题:gluon.Trainer()

      本文链接:https://www.haomeiwen.com/subject/gsrkshtx.html