2019-03-31 01:45:16 +00:00
|
|
|
import torch
|
|
|
|
|
|
|
|
|
|
class ModelHooks(torch.nn.Module):
|
2019-04-21 17:40:32 +00:00
|
|
|
def on_batch_start(self, data_batch):
|
2019-03-31 01:45:16 +00:00
|
|
|
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
|
2019-04-21 16:26:35 +00:00
|
|
|
|
2019-07-16 16:51:48 +00:00
|
|
|
def on_tng_metrics(self, metrics):
|
|
|
|
|
pass
|
|
|
|
|
|
