(rontolisp) docs
← 関数

torch:step

(torch:step optimizer)

オプティマイザの更新則をすべてのパラメータに適用し、オプティマイザを返します (PyTorch の optimizer.step())。ステップカウンタは先に加算されるので、torch:step-count を読むバイアス補正は最初のステップで 1 を見ます。

更新は torch の演算を使わず各パラメータのデータを直接書き換えるため、テープには何も記録されません。torch:set-data で手書きした更新と違い、torch:no-grad で囲む必要はありません。勾配がまだ NIL のパラメータは飛ばされます。