lib/torch/optim/rprop.rb in torch-rb-0.2.5 vs lib/torch/optim/rprop.rb in torch-rb-0.2.6
- old
+ new
@@ -9,12 +9,9 @@
defaults = {lr: lr, etas: etas, step_sizes: step_sizes}
super(params, defaults)
end
def step(closure = nil)
- # TODO implement []=
- raise NotImplementedYet
-
loss = nil
if closure
loss = closure.call
end