nntoolbox.callbacks.debug module

Implement a debug callback. Adapt from fastai course2 v3 notebook 11 a

class nntoolbox.callbacks.debug.DebugCallback(step_to_debug: str, func)

Bases: nntoolbox.callbacks.callbacks.Callback

after_backward() → bool

after_losses(losses: Dict[str, torch.Tensor], train: bool) → Dict[str, torch.Tensor]

after_outputs(outputs: Dict[str, torch.Tensor], train: bool) → Dict[str, torch.Tensor]

after_step() → bool

on_backward_begin() → bool

on_batch_begin(data: Dict[str, torch.Tensor], train) → Dict[str, torch.Tensor]

on_batch_end(logs: Dict[str, Any])

on_epoch_begin()

on_epoch_end(logs: Dict[str, Any]) → bool

on_train_begin()

on_train_end()