unilab.base.curriculum.EpisodeLengthTracker
-
class unilab.base.curriculum.EpisodeLengthTracker[source]
Bases: object
Track moving average of episode length.
- Parameters:
num_envs (int)
window_size (int)
Methods
__init__(num_envs[, window_size])
|
|
update(episode_lengths)
|
Update average with new episode lengths. |
-
__init__(num_envs, window_size=1000)[source]
- Parameters:
num_envs (int)
window_size (int)
-
update(episode_lengths)[source]
Update average with new episode lengths.
- Parameters:
episode_lengths (ndarray)
- Return type:
None