test_optimizer.py 文件源码

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

项目:chainer-deconv 作者: germanRos 项目源码 文件源码
def check_weight_decay(self):
        w = self.target.param.data
        g = self.target.param.grad

        decay = 0.2
        expect = w - g - decay * w

        opt = optimizers.SGD(lr=1)
        opt.setup(self.target)
        opt.add_hook(optimizer.WeightDecay(decay))
        opt.update()

        gradient_check.assert_allclose(expect, w)
评论列表
文章目录


问题


面经


文章

微信
公众号

扫码关注公众号