2020-01-28 20:07:55 +01:00
|
|
|
from ray.rllib.utils.schedules.polynomial_schedule import PolynomialSchedule
|
|
|
|
|
|
|
|
|
|
|
|
class LinearSchedule(PolynomialSchedule):
|
2021-12-15 22:32:52 +01:00
|
|
|
"""Linear interpolation between `initial_p` and `final_p`.
|
|
|
|
|
|
|
|
Uses `PolynomialSchedule` with power=1.0.
|
|
|
|
|
2020-01-28 20:07:55 +01:00
|
|
|
|
2021-12-15 22:32:52 +01:00
|
|
|
The formula is:
|
|
|
|
value = `final_p` + (`initial_p` - `final_p`) * (1 - `t`/t_max)
|
2020-01-28 20:07:55 +01:00
|
|
|
"""
|
|
|
|
|
|
|
|
def __init__(self, **kwargs):
|
2021-12-15 22:32:52 +01:00
|
|
|
"""Initializes a LinearSchedule instance."""
|
2020-01-28 20:07:55 +01:00
|
|
|
super().__init__(power=1.0, **kwargs)
|