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

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

用法:

static 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.Optimizer.register。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。