topobench.callbacks package#
Submodules#
topobench.callbacks.timer_callback module#
TopoBench: A library for benchmarking topological models.
- class topobench.callbacks.timer_callback.PipelineTimer[source]#
Bases:
CallbackMeasures and logs average execution times of training, validation, and testing stages.
- on_test_batch_end(*args)[source]#
End timing a test batch.
- Parameters:
- *argstuple
Additional arguments passed by the trainer.
- on_test_batch_start(*args)[source]#
Start timing a test batch.
- Parameters:
- *argstuple
Additional arguments passed by the trainer.
- on_test_epoch_end(*args)[source]#
End timing a test epoch.
- Parameters:
- *argstuple
Additional arguments passed by the trainer.
- on_test_epoch_start(*args)[source]#
Start timing a test epoch.
- Parameters:
- *argstuple
Additional arguments passed by the trainer.
- on_train_batch_end(*args)[source]#
End timing a training batch.
- Parameters:
- *argstuple
Additional arguments passed by the trainer.
- on_train_batch_start(*args)[source]#
Start timing a training batch.
- Parameters:
- *argstuple
Additional arguments passed by the trainer.
- on_train_end(trainer, *args)[source]#
Log the average times at the end of training.
- Parameters:
- trainerobject
The PyTorch Lightning trainer instance used for logging.
- *argstuple
Additional arguments passed by the trainer.
- on_train_epoch_end(*args)[source]#
End timing a training epoch.
- Parameters:
- *argstuple
Additional arguments passed by the trainer.
- on_train_epoch_start(*args)[source]#
Start timing a training epoch.
- Parameters:
- *argstuple
Additional arguments passed by the trainer.
- on_validation_batch_end(*args)[source]#
End timing a validation batch.
- Parameters:
- *argstuple
Additional arguments passed by the trainer.
- on_validation_batch_start(*args)[source]#
Start timing a validation batch.
- Parameters:
- *argstuple
Additional arguments passed by the trainer.
Module contents#
TopoBench: A library for benchmarking of topological models.