Class ReduceLROnPlateauScheduler¶
Defined in File reduce_on_plateau_scheduler.h
Page Contents
Class Documentation¶
-
class ReduceLROnPlateauScheduler¶
Public Types
Public Functions
-
ReduceLROnPlateauScheduler(Optimizer &optimizer, SchedulerMode mode = min, float factor = 0.1, int patience = 10, double threshold = 1e-4, ThresholdMode threshold_mode = rel, int cooldown = 0, const std::vector<float> &min_lr = std::vector<float>(), double eps = 1e-8, bool verbose = false)¶
-
virtual ~ReduceLROnPlateauScheduler() = default¶
-
void step(float metric)¶
-
ReduceLROnPlateauScheduler(Optimizer &optimizer, SchedulerMode mode = min, float factor = 0.1, int patience = 10, double threshold = 1e-4, ThresholdMode threshold_mode = rel, int cooldown = 0, const std::vector<float> &min_lr = std::vector<float>(), double eps = 1e-8, bool verbose = false)¶