Searched defs:pl_module (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/ao/pruning/_experimental/data_sparsifier/lightning/callbacks/ |
| H A 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
157 def __create_config_based_on_state(self, pl_module): argument
167 def on_train_epoch_end(self, trainer, pl_module): argument
180 def on_train_end(self, trainer, pl_module): argument
|
| /aosp_15_r20/external/pytorch/torch/ao/pruning/_experimental/data_sparsifier/lightning/tests/ |
| H A D | test_callbacks.py | 93 def _check_on_fit_end(self, pl_module, callback, sparsifier_args): argument
152 self, pl_module, callback, sparsifier_args, scheduler_args argument
174 def _simulate_update_param_model(self, pl_module): argument
181 def _check_on_train_epoch_start(self, pl_module, callback): argument
241 def _check_on_train_epoch_end(self, pl_module, callback): argument
267 def _check_on_train_end(self, pl_module, callback): argument
|