sgd_step: Sgd_step

Description Usage Arguments Value Examples

View source: R/optimizers.R

Description

Sgd_step

Usage

1
sgd_step(p, lr, ...)

Arguments

p

p

lr

learning rate

...

additional arguments to pass

Value

None

Examples

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
## Not run: 

tst_param = function(val, grad = NULL) {
  "Create a tensor with `val` and a gradient of `grad` for testing"
  res = tensor(val) %>% float()

  if(is.null(grad)) {
    grad = tensor(val / 10)
  } else {
    grad = tensor(grad)
  }

  res$grad = grad %>% float()
  res
}
p = tst_param(1., 0.1)
sgd_step(p, 1.)


## End(Not run)

fastai documentation built on Oct. 25, 2021, 5:08 p.m.