当前位置: 首页>>编程示例 >>用法及示例精选 >>正文


Python mxnet.optimizer.register用法及代码示例

用法:

mxnet.optimizer.register(klass)

注册一个新的优化器。

注册优化器后,我们可以稍后使用create_optimizer 创建此优化器的实例。

例子

>>> @mx.optimizer.Optimizer.register
... class MyOptimizer(mx.optimizer.Optimizer):
...     pass
>>> optim = mx.optimizer.Optimizer.create_optimizer('MyOptimizer')
>>> print(type(optim))
<class '__main__.MyOptimizer'>

相关用法


注:本文由纯净天空筛选整理自apache.org大神的英文原创作品 mxnet.optimizer.register。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。