linear_schedule.py 528 B

123456789101112131415161718
  1. from ray.rllib.utils.annotations import OldAPIStack
  2. from ray.rllib.utils.schedules.polynomial_schedule import PolynomialSchedule
  3. @OldAPIStack
  4. class LinearSchedule(PolynomialSchedule):
  5. """Linear interpolation between `initial_p` and `final_p`.
  6. Uses `PolynomialSchedule` with power=1.0.
  7. The formula is:
  8. value = `final_p` + (`initial_p` - `final_p`) * (1 - `t`/t_max)
  9. """
  10. def __init__(self, **kwargs):
  11. """Initializes a LinearSchedule instance."""
  12. super().__init__(power=1.0, **kwargs)