law sgd_no_params

bendtorch-core@0.1.0.0 · src/autograd/sgd_LAWS.bend:108

import Base
import bendtorch-core@0.1.0.0/src/autograd/tensor_backward.bend as Backward
import bendtorch-core@0.1.0.0/src/autograd/sgd.bend as Sgd

108# The parameters are updated in list order: none leaves the entries as they
109# are, and the first is updated before the rest, which see its result.
110law sgd_no_params:
111  for +lr: F32
112  for +count: Nat
113  for entries: List<&1, Backward.Entry>
114  {Sgd.descend_all(lr, count, Nil{}, entries) == (entries, Done{Unit{}}) : Sgd.Walked()}