unilab.base.curriculum

Curriculum learning for adaptive difficulty adjustment.

Classes

EpisodeLengthTracker

Track moving average of episode length.
class unilab.base.curriculum.EpisodeLengthTracker[source]

Bases: object

Track moving average of episode length.

Parameters:

  • num_envs (int)

  • window_size (int)

__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