ray/rllib/utils/schedules/linear_schedule.py

Ignoring revisions in .git-blame-ignore-revs. Click here to bypass and see the normal blame view.

19 lines
524 B
Python
Raw Normal View History

from ray.rllib.utils.annotations import PublicAPI
from ray.rllib.utils.schedules.polynomial_schedule import PolynomialSchedule
@PublicAPI
class LinearSchedule(PolynomialSchedule):
"""Linear interpolation between `initial_p` and `final_p`.
Uses `PolynomialSchedule` with power=1.0.
The formula is:
value = `final_p` + (`initial_p` - `final_p`) * (1 - `t`/t_max)
"""
def __init__(self, **kwargs):
"""Initializes a LinearSchedule instance."""
super().__init__(power=1.0, **kwargs)