obtain_normalization_statistics

class maze.core.wrappers.observation_normalization.observation_normalization_utils.obtain_normalization_statistics(env: Union[maze.core.env.maze_env.MazeEnv, maze.core.wrappers.observation_normalization.observation_normalization_wrapper.ObservationNormalizationWrapper], n_samples: int)

Obtain the normalization statistics of a given environment.

• Returns None, if the ObservationNormalizationWrapper is not implemented

• Returns the loaded statistics, if available

• Runs the estimation and returns the newly calculated statistics, if not loaded previously

Parameters

• env – Environment with applied ObservationNormalizationWrapper (function returns None immediately if this is not the case.

• n_samples – Number of samples (=steps) to collect normalization statistics at the beginning of the training.

Returns

The normalization statistics or None if the ObservationNormalizationWrapper is not implemented by the env.