mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-10-27 13:12:20 +08:00
24 lines
396 B
Python
24 lines
396 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
|
|
|
|
def should_stop_epoch(self, data_batch):
|
|
return False
|