unilab.ipc.shared_obs_stats

Shared observation normalization statistics for multi-process training.

Classes

SharedObsNormStats

Synchronize observation normalization statistics between learner and collector.

class unilab.ipc.shared_obs_stats.SharedObsNormStats[source]

Bases: object

Synchronize observation normalization statistics between learner and collector.

Uses a queue to pass (mean, std) tuples from learner to collector.

__init__(ctx)[source]
put(stats)[source]

Put new stats, clearing old ones first.

get()[source]

Get latest stats, returns None if no new stats.