| optimizer_sgd | R Documentation |
Create an SGD optimizer
optimizer_sgd(params, lr = 0.01, momentum = 0)
params |
Named list of ag_param tensors |
lr |
Learning rate (default 0.01) |
momentum |
Momentum factor (default 0) |
An optimizer environment
w <- ag_param(matrix(runif(4), 2, 2))
opt <- optimizer_sgd(list(w = w), lr = 0.01)
Add the following code to your website.
For more information on customizing the embed code, read Embedding Snippets.