當前位置: 首頁>>編程示例 >>用法及示例精選 >>正文

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。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。