import torch class ModelHooks(torch.nn.Module): def on_batch_start(self): pass def on_batch_end(self): pass def on_epoch_start(self): pass def on_epoch_end(self): pass def on_pre_performance_check(self): pass def on_post_performance_check(self): pass