美文网首页
固定部分参数,学习率

固定部分参数,学习率

作者: 发现一个喜悦的地方 | 来源:发表于2020-02-14 13:55 被阅读0次

    1.固定网络参数

    转自https://blog.csdn.net/qq_21997625/article/details/90369838

    for i,p in enumerate(net.parameters()):
    if i < 165:
    p.requires_grad = False

    2.调整学习率

    http://www.spytensor.com/index.php/archives/32/

    (https://zhuanlan.zhihu.com/p/93624972
    pytorch必须掌握的的4种学习率衰减策略

    '''
    scheduler = MultiStepLR(optimizer, milestones=[30,80], gamma=0.1)
    for epoch in range(100):
    train(...)
    validate(...)
    scheduler.step()
    '''

    相关文章

      网友评论

          本文标题:固定部分参数,学习率

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