lib/dnn/core/model.rb in ruby-dnn-0.9.2 vs lib/dnn/core/model.rb in ruby-dnn-0.9.3
- old
+ new
@@ -254,10 +254,10 @@
x, y = batch_proc.call(x, y) if batch_proc
out = forward(x, true)
loss_value = @loss.forward(out, y, get_all_layers)
dout = @loss.backward(y)
backward(dout)
- @loss.d_regularize(get_all_layers)
+ @loss.regularizes_backward(get_all_layers)
update
loss_value
end
# Evaluate model and get accurate of test data.