def learning_rate_scheduler(epoch, values=(0.1, 0.01, 0.001),
                            breakpoints=(100, 150)):
  """Piecewise constant schedule for learning rate."""
  idx = sum(1 if epoch > b else 0 for b in breakpoints)
  return values[idx]