mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-09-21 13:20:08 +08:00
Alok Singh
b0a0a47a0b
- data_batch → batch - batch_i → batch_idx - dataloader_i → dataloader_idx - tng → training - training_dataloader → train_dataloader - add_log_row_interval → row_log_interval - gradient_clip → gradient_clip_val - prog → progress - tqdm_dic → tqdm_dict
54 lines
1.0 KiB
Python
54 lines
1.0 KiB
Python
import torch
|
|
|
|
|
|
class ModelHooks(torch.nn.Module):
|
|
|
|
def on_sanity_check_start(self):
|
|
"""
|
|
Called before starting evaluate
|
|
:return:
|
|
"""
|
|
pass
|
|
|
|
def on_batch_start(self, batch):
|
|
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 on_training_metrics(self, metrics):
|
|
pass
|
|
|
|
def on_before_zero_grad(self, optimizer):
|
|
"""
|
|
Called after optimizer.step() and before optimizer.zero_grad()
|
|
|
|
for optimizer in optimizers:
|
|
optimizer.step()
|
|
model.on_before_zero_grad(optimizer) # < ---- called here
|
|
optimizer.zero_grad
|
|
|
|
:param optimizer:
|
|
:return:
|
|
"""
|
|
pass
|
|
|
|
def on_after_backward(self):
|
|
"""
|
|
Called after loss.backward() and before optimizers do anything
|
|
:return:
|
|
"""
|
|
pass
|