mirror of
https://github.com/vale981/ray
synced 2025-03-09 04:46:38 -04:00
14 lines
376 B
Python
14 lines
376 B
Python
|
from ray.rllib.utils.schedules.polynomial_schedule import PolynomialSchedule
|
||
|
|
|
||
|
|
|
||
|
|
class LinearSchedule(PolynomialSchedule):
|
||
|
|
"""
|
||
|
|
Linear interpolation between `initial_p` and `final_p`. Simply
|
||
|
|
uses Polynomial with power=1.0.
|
||
|
|
|
||
|
|
final_p + (initial_p - final_p) * (1 - `t`/t_max)
|
||
|
|
"""
|
||
|
|
|
||
|
|
def __init__(self, **kwargs):
|
||
|
|
super().__init__(power=1.0, **kwargs)
|