lrAt method

@override double lrAt(int step)

The LR that will be applied on the next step() call.

Implementation

@override
double lrAt(int step) {
  final k = step ~/ stepSize;
  var lr = initialLr;
  for (int i = 0; i < k; i++) {
    lr *= gamma;
  }
  return lr;
}