espnet.asr.asr_utils.torch_snapshot

source

espnet.asr.asr_utils.torch_snapshot(savefun=<function save>, filename='snapshot.ep.{.updater.epoch}')

Extension to take snapshot of the trainer for pytorch.

• Returns: An extension function.