lr_scheduler: Creates learning rate schedulers

Description Usage Arguments

View source: R/optim-lr_scheduler.R

Description

Creates learning rate schedulers

Usage

1
2
3
4
5
6
lr_scheduler(
  classname = NULL,
  inherit = LRScheduler,
  ...,
  parent_env = parent.frame()
)

Arguments

classname

optional name for the learning rate scheduler

inherit

an optional learning rate scheduler to inherit from

...

named list of methods. You must implement the get_lr() method that doesn't take any argument and returns learning rates for each param_group in the optimizer.

parent_env

passed to R6::R6Class().


torch documentation built on Oct. 7, 2021, 9:22 a.m.