def _build_optimizer(self, config):
kwargs = {k: float(v) for k, v in config.items() if k != 'name'}
o = getattr(chainer.optimizers, config['name'])(**kwargs)
return o
微信
公众号
扫码关注公众号