nutszebra_optimizer.py 文件源码

python
阅读 19 收藏 0 点赞 0 评论 0

项目:googlenet_v3 作者: nutszebra 项目源码 文件源码
def __init__(self, model=None, lr=0.045, momentum=0.9, weight_decay=4.0e-5):
        super(OptimizerGooglenetV2, self).__init__(model)
        optimizer = optimizers.MomentumSGD(lr, momentum)
        weight_decay = chainer.optimizer.WeightDecay(weight_decay)
        optimizer.setup(self.model)
        optimizer.add_hook(weight_decay)
        self.optimizer = optimizer
        self.lr = lr
        self.momentum = momentum
        self.weight_decay = weight_decay
评论列表
文章目录


问题


面经


文章

微信
公众号

扫码关注公众号