disable_epoch_level_stats

class maze.train.parallelization.vector_env.vector_env_utils.disable_epoch_level_stats(env: LogStatsWrapper)

Disable collection of statistics on epoch level to save memory.

This is useful in distributed vectorized env scenarios as episode statistics are shipped to the main worker/node and there is no need to keep them and aggregate them on worker level.