Searched defs:trainer (Results 1 – 1 of 1) sorted by relevance
| /external/pytorch/torch/ao/pruning/_experimental/data_sparsifier/lightning/callbacks/ |
| D | data_sparsity.py | 53 def on_fit_end(self, trainer, pl_module) -> None: argument
136 def on_train_start(self, trainer, pl_module) -> None: argument
150 def on_train_epoch_start(self, trainer, pl_module): argument
167 def on_train_epoch_end(self, trainer, pl_module): argument
180 def on_train_end(self, trainer, pl_module): argument
|