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


mlr3torch documentation built on April 4, 2025, 3:03 a.m.