speechbrain.utils.epoch_loop module

Implements a checkpointable epoch counter (loop).

Authors

• Aku Rouhe 2020

Summary

Classes:

EpochCounter

An epoch counter which can save and recall its state.

Reference

class speechbrain.utils.epoch_loop.EpochCounter(limit)

Bases: object

An epoch counter which can save and recall its state.

Use this as the iterator for epochs. Note that this iterator gives you the numbers from [1 … limit] not [0 … limit-1] as range(limit) would.

Example

>>> from speechbrain.utils.checkpoints import Checkpointer
>>> tmpdir = getfixture('tmpdir')
>>> epoch_counter = EpochCounter(10)
>>> recoverer = Checkpointer(tmpdir, {"epoch": epoch_counter})
>>> recoverer.recover_if_possible()
>>> # Now after recovery,
>>> # the epoch starts from where it left off!
>>> for epoch in epoch_counter:
...     # Run training...
...     ckpt = recoverer.save_checkpoint()