as_lr_scheduler: Convert to CallbackSetLRScheduler

View source: R/CallbackSetLRScheduler.R

as_lr_schedulerR Documentation

Convert to CallbackSetLRScheduler

Description

Convert a torch scheduler generator to a CallbackSetLRScheduler.

Usage

as_lr_scheduler(x, step_on_epoch)

Arguments

x

(function)
The torch scheduler generator defined using torch::lr_scheduler().

step_on_epoch

(logical(1))
Whether the scheduler steps after every epoch


mlr-org/mlr3torch documentation built on April 17, 2025, 8:22 p.m.