Skip to content

Curiosity

Curiosity

Intrinsic Curiosity Module (ICM) based agent.

CuriosityTrainer

CuriosityTrainer(envs, agent: Curiosity, val_envs, config: TrainerConfig)

Bases: SyncMultiEnvTrainer

Trainer for the Intrinsic Curiosity Module agent.

Source code in rlib/Curiosity/trainer.py
30
31
32
33
34
35
36
37
38
39
def __init__(
    self,
    envs,
    agent: Curiosity,
    val_envs,
    config: TrainerConfig,
):
    super().__init__(envs, agent, val_envs, config=config)
    self.state_obs = RollingObs()
    self.lambda_ = 0.95