load_rng_state#

composer.utils.reproducibility.load_rng_state(rng_state_dicts)[source]#

Restore the RNG state.

Parameters

rng_state_dicts (list[dict[str, Any]]) โ€“ The list of RNG state dicts to restore, as returned by get_rng_state().