Home
last modified time | relevance | path

Searched defs:on_train_end (Results 1 – 3 of 3) sorted by relevance

/external/pytorch/torch/ao/pruning/_experimental/data_sparsifier/lightning/callbacks/
Ddata_sparsity.py180 def on_train_end(self, trainer, pl_module): member in TrainingAwareDataSparsity
/external/tensorflow/tensorflow/python/keras/
Dcallbacks.py513 def on_train_end(self, logs=None): member in CallbackList
830 def on_train_end(self, logs=None): member in Callback
1695 def on_train_end(self, logs=None): member in BackupAndRestore
1837 def on_train_end(self, logs=None): member in EarlyStopping
2403 def on_train_end(self, logs=None): member in TensorBoard
2800 def on_train_end(self, logs=None): member in CSVLogger
2864 on_train_end=None, argument
Dcallbacks_v1.py458 def on_train_end(self, logs=None): member in TensorBoard