components
Submodules
- callback_handler
CallbacksHandler
CallbacksHandler.register_callbacks()
CallbacksHandler.should_enable_callback()
CallbacksHandler.execute_epoch_begin()
CallbacksHandler.execute_epoch_end()
CallbacksHandler.execute_train_begin()
CallbacksHandler.execute_train_end()
CallbacksHandler.execute_batch_begin()
CallbacksHandler.execute_batch_end()
CallbacksHandler.execute_gradient_update()
CallbacksHandler.execute_optimizer_step()
CallbacksHandler.execute_multiprocess_start()
CallbacksHandler.execute_after_batch_prediction()
CallbacksHandler.split_on_execution_position()
CallbacksHandler.mp_filter_callbacks()
CallbacksHandler.enforce_callbacks_quality()
CallbacksHandler.print_callback_info()
CallbacksHandler.print_registered_callback_names()
CallbacksHandler.__add__()
CallbacksHandler.__iadd__()
CallbacksHandler.__contains__()
- ddp_handler
- message_passing
- model_prediction_store
ModelPredictionStore
ModelPredictionStore.insert_train_predictions()
ModelPredictionStore.insert_val_predictions()
ModelPredictionStore.insert_test_predictions()
ModelPredictionStore.get_train_predictions()
ModelPredictionStore.get_val_predictions()
ModelPredictionStore.get_test_predictions()
ModelPredictionStore.has_train_predictions()
ModelPredictionStore.has_val_predictions()
ModelPredictionStore.has_test_predictions()
ModelPredictionStore.insert_train_loss()
ModelPredictionStore.insert_val_loss()
ModelPredictionStore.insert_test_loss()
ModelPredictionStore.get_train_loss()
ModelPredictionStore.get_val_loss()
ModelPredictionStore.get_test_loss()
ModelPredictionStore.has_train_loss()
ModelPredictionStore.has_val_loss()
ModelPredictionStore.has_test_loss()
ModelPredictionStore._insert_data()
ModelPredictionStore._get_data()
ModelPredictionStore._has_data()
ModelPredictionStore.auto_purge()
- pred_collate_fns