espnet2.train.lightning_callbacks.AverageCheckpointsCallback

source

class espnet2.train.lightning_callbacks.AverageCheckpointsCallback(output_dir, best_ckpt_callbacks)

Bases: Callback

on_fit_end(trainer, pl_module)

Called when fit ends.