21 lines
329 B
Python
21 lines
329 B
Python
|
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
|