美文网首页
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()

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