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


Python mxnet.optimizer.create用法及代碼示例


用法:

mxnet.optimizer.create(name, **kwargs)

參數

  • name(str) - 優化器的名稱。應該是優化器子類的名稱。不區分大小寫。
  • kwargs(dict) - 優化器的參數。

返回

一個實例化的優化器。

返回類型

優化器

實例化具有給定名稱和 kwargs 的優化器。

注意

我們可以為 Optimizer.create_optimizer 使用別名 create

例子

>>> sgd = mx.optimizer.Optimizer.create_optimizer('sgd')
>>> type(sgd)
<class 'mxnet.optimizer.SGD'>
>>> adam = mx.optimizer.create('adam', learning_rate=.1)
>>> type(adam)
<class 'mxnet.optimizer.Adam'>

相關用法


注:本文由純淨天空篩選整理自apache.org大神的英文原創作品 mxnet.optimizer.create。非經特殊聲明,原始代碼版權歸原作者所有,本譯文未經允許或授權,請勿轉載或複製。