Skip to content

RewardCallback

Full path: schola.scripts.sb3.utils.RewardCallback

Callback for logging rewards and steps taken by each environment in a multi-env setting.

RewardCallback(verbose = 0, frequency = 10, num_envs = 1)

Parameters

  • verbose (int, default: 0) - Verbosity level.

  • frequency (int, default: 10) - The frequency at which to log the rewards and steps taken.

  • num_envs (int, default: 1) - The number of environments to log rewards and steps for.

Methods

init

__init__(verbose = 0, frequency = 10, num_envs = 1)

Parameters

  • verbose (int)

  • frequency (int)

  • num_envs (int)

Attributes

num_envs

num_envs : int

The number of environments to log rewards and steps for.

callbacks

callbacks : List[SingleEnvRewardCallback]

The list of RewardLoggingCallbacks for each environment.

summarize_every

summarize_every : int

The frequency at which to log the rewards and steps taken.

curr_logging_interval

curr_logging_interval : int

The current logging interval.

start_time

start_time : int

The time at which the callback was created.

ready_to_log

ready_to_log