components
Submodules
- callback_handler
CallbacksHandlerCallbacksHandler.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
ModelPredictionStoreModelPredictionStore.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