mirror of https://github.com/coqui-ai/TTS.git
102 lines
4.0 KiB
Python
102 lines
4.0 KiB
Python
class TrainerCallback:
|
|
def __init__(self):
|
|
super().__init__()
|
|
|
|
def on_init_start(self, trainer) -> None:
|
|
if hasattr(trainer.model, "module"):
|
|
if hasattr(trainer.model.module, "on_init_start"):
|
|
trainer.model.module.on_init_start(trainer)
|
|
else:
|
|
if hasattr(trainer.model, "on_init_start"):
|
|
trainer.model.on_init_start(trainer)
|
|
|
|
if hasattr(trainer.criterion, "on_init_start"):
|
|
trainer.criterion.on_init_start(trainer)
|
|
|
|
if hasattr(trainer.optimizer, "on_init_start"):
|
|
trainer.optimizer.on_init_start(trainer)
|
|
|
|
def on_init_end(self, trainer) -> None:
|
|
if hasattr(trainer.model, "module"):
|
|
if hasattr(trainer.model.module, "on_init_end"):
|
|
trainer.model.module.on_init_end(trainer)
|
|
else:
|
|
if hasattr(trainer.model, "on_init_end"):
|
|
trainer.model.on_init_end(trainer)
|
|
|
|
if hasattr(trainer.criterion, "on_init_end"):
|
|
trainer.criterion.on_init_end(trainer)
|
|
|
|
if hasattr(trainer.optimizer, "on_init_end"):
|
|
trainer.optimizer.on_init_end(trainer)
|
|
|
|
def on_epoch_start(self, trainer) -> None:
|
|
if hasattr(trainer.model, "module"):
|
|
if hasattr(trainer.model.module, "on_epoch_start"):
|
|
trainer.model.module.on_epoch_start(trainer)
|
|
else:
|
|
if hasattr(trainer.model, "on_epoch_start"):
|
|
trainer.model.on_epoch_start(trainer)
|
|
|
|
if hasattr(trainer.criterion, "on_epoch_start"):
|
|
trainer.criterion.on_epoch_start(trainer)
|
|
|
|
if hasattr(trainer.optimizer, "on_epoch_start"):
|
|
trainer.optimizer.on_epoch_start(trainer)
|
|
|
|
def on_epoch_end(self, trainer) -> None:
|
|
if hasattr(trainer.model, "module"):
|
|
if hasattr(trainer.model.module, "on_epoch_end"):
|
|
trainer.model.module.on_epoch_end(trainer)
|
|
else:
|
|
if hasattr(trainer.model, "on_epoch_end"):
|
|
trainer.model.on_epoch_end(trainer)
|
|
|
|
if hasattr(trainer.criterion, "on_epoch_end"):
|
|
trainer.criterion.on_epoch_end(trainer)
|
|
|
|
if hasattr(trainer.optimizer, "on_epoch_end"):
|
|
trainer.optimizer.on_epoch_end(trainer)
|
|
|
|
def on_train_step_start(self, trainer) -> None:
|
|
if hasattr(trainer.model, "module"):
|
|
if hasattr(trainer.model.module, "on_train_step_start"):
|
|
trainer.model.module.on_train_step_start(trainer)
|
|
else:
|
|
if hasattr(trainer.model, "on_train_step_start"):
|
|
trainer.model.on_train_step_start(trainer)
|
|
|
|
if hasattr(trainer.criterion, "on_train_step_start"):
|
|
trainer.criterion.on_train_step_start(trainer)
|
|
|
|
if hasattr(trainer.optimizer, "on_train_step_start"):
|
|
trainer.optimizer.on_train_step_start(trainer)
|
|
|
|
def on_train_step_end(self, trainer) -> None:
|
|
if hasattr(trainer.model, "module"):
|
|
if hasattr(trainer.model.module, "on_train_step_end"):
|
|
trainer.model.module.on_train_step_end(trainer)
|
|
else:
|
|
if hasattr(trainer.model, "on_train_step_end"):
|
|
trainer.model.on_train_step_end(trainer)
|
|
|
|
if hasattr(trainer.criterion, "on_train_step_end"):
|
|
trainer.criterion.on_train_step_end(trainer)
|
|
|
|
if hasattr(trainer.optimizer, "on_train_step_end"):
|
|
trainer.optimizer.on_train_step_end(trainer)
|
|
|
|
def on_keyboard_interrupt(self, trainer) -> None:
|
|
if hasattr(trainer.model, "module"):
|
|
if hasattr(trainer.model.module, "on_keyboard_interrupt"):
|
|
trainer.model.module.on_keyboard_interrupt(trainer)
|
|
else:
|
|
if hasattr(trainer.model, "on_keyboard_interrupt"):
|
|
trainer.model.on_keyboard_interrupt(trainer)
|
|
|
|
if hasattr(trainer.criterion, "on_keyboard_interrupt"):
|
|
trainer.criterion.on_keyboard_interrupt(trainer)
|
|
|
|
if hasattr(trainer.optimizer, "on_keyboard_interrupt"):
|
|
trainer.optimizer.on_keyboard_interrupt(trainer)
|