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