Skip to contents

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