StepLR constructor

StepLR(
  1. Optimizer optimizer, {
  2. required double initialLr,
  3. required int stepSize,
  4. double gamma = 0.1,
})

Implementation

StepLR(
  super.optimizer, {
  required this.initialLr,
  required this.stepSize,
  this.gamma = 0.1,
}) {
  if (stepSize <= 0) {
    throw ArgumentError('StepLR: stepSize must be > 0');
  }
  optimizer.lr = initialLr;
}