The grads tuple
will correspond to the same order dnn_Linear.parameters()
is returning the parameters, so you could iterate both and apply your update rule.
The grads tuple
will correspond to the same order dnn_Linear.parameters()
is returning the parameters, so you could iterate both and apply your update rule.