core.trainers.mixins.rng_state_loader

core.trainers.mixins.rng_state_loader

Temporary fix/override for bug in resume from checkpoint

See https://github.com/huggingface/transformers/pull/37162

TODO: Remove when upstream added PR to release

Classes

Name Description
RngLoaderMixin mixin for method override to load RNG states from a checkpoint

RngLoaderMixin

core.trainers.mixins.rng_state_loader.RngLoaderMixin()

mixin for method override to load RNG states from a checkpoint