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