(rontolisp) docs
← 関数

torch:optimizer

(torch:optimizer kind params fields step-fn)

新しいオプティマイザを返します。kind は更新則を表すキーワード、params はモジュール (torch:parameters をたどります) またはパラメータテンソルのリスト、fields はハイパーパラメータや内部状態を保持する KEYWORD/値の plist、step-fntorch:step(funcall step-fn optimizer) として呼ぶ関数です。ステップカウンタは 0 から始まります。

ユーザー定義の更新則はこの形で書きます。torch:sgdtorch:adam も単にこれを呼ぶだけです。ステップ関数はハイパーパラメータを torch:field で、パラメータを torch:optimizer-params で読み戻すので、更新則が必要とするものはクロージャではなくレコードの中に置かれます。