callbacks

Submodules

• abstract
  • AbstractCallback
    • AbstractCallback.register_train_loop_object()
    • AbstractCallback.on_train_loop_registration()
    • AbstractCallback.on_epoch_begin()
    • AbstractCallback.on_epoch_end()
    • AbstractCallback.on_train_begin()
    • AbstractCallback.on_train_end()
    • AbstractCallback.on_batch_begin()
    • AbstractCallback.on_batch_end()
    • AbstractCallback.on_after_gradient_update()
    • AbstractCallback.on_after_optimizer_step()
    • AbstractCallback.on_multiprocess_start()
    • AbstractCallback.on_after_batch_prediction()
  • AbstractExperimentCallback
    • AbstractExperimentCallback.try_infer_experiment_details()
• basic
  • ListRegisteredCallbacks
    • ListRegisteredCallbacks.on_train_begin()
  • EarlyStopping
    • EarlyStopping.on_epoch_end()
  • ThresholdEarlyStopping
    • ThresholdEarlyStopping.on_epoch_end()
  • TerminateOnNaN
    • TerminateOnNaN.on_epoch_end()
  • AllPredictionsSame
    • AllPredictionsSame.on_epoch_end()
  • EmailNotification
    • EmailNotification.on_epoch_end()
    • EmailNotification.on_train_end()
    • EmailNotification.get_metric_list_html()
    • EmailNotification.get_hyperparams_html()
    • EmailNotification.on_train_loop_registration()
  • LogUpload
    • LogUpload.on_train_loop_registration()
    • LogUpload.on_epoch_end()
    • LogUpload.on_train_end()
    • LogUpload.upload_log_file()
  • DataSubsetTestRun
    • DataSubsetTestRun.on_train_begin()
    • DataSubsetTestRun.subset_data_loader()
    • DataSubsetTestRun.on_train_loop_registration()
  • FunctionOnTrainLoop
    • FunctionOnTrainLoop.execute_callback()
    • FunctionOnTrainLoop.on_train_loop_registration()
    • FunctionOnTrainLoop.on_epoch_begin()
    • FunctionOnTrainLoop.on_epoch_end()
    • FunctionOnTrainLoop.on_train_begin()
    • FunctionOnTrainLoop.on_train_end()
    • FunctionOnTrainLoop.on_batch_begin()
    • FunctionOnTrainLoop.on_batch_end()
    • FunctionOnTrainLoop.on_after_gradient_update()
    • FunctionOnTrainLoop.on_after_optimizer_step()
• ddp
  • DistributedSamplerSetEpoch
    • DistributedSamplerSetEpoch.on_epoch_begin()
  • InMultiProcessDataLoad
    • InMultiProcessDataLoad.on_multiprocess_start()
    • InMultiProcessDataLoad.build_train_dataloader()
    • InMultiProcessDataLoad.build_val_dataloader()
    • InMultiProcessDataLoad.build_test_dataloader()
• gradient
  • GradientCallbackBase
    • GradientCallbackBase.on_train_loop_registration()
  • GradValueClip
    • GradValueClip.on_after_gradient_update()
  • GradNormClip
    • GradNormClip.on_after_gradient_update()
  • GradientStatsPrint
    • GradientStatsPrint.on_after_gradient_update()
    • GradientStatsPrint.gradients_report()
  • GradDistributionPlot
    • GradDistributionPlot.on_train_loop_registration()
    • GradDistributionPlot.on_epoch_end()
    • GradDistributionPlot.gradient_plot()
    • GradDistributionPlot.save_to_cloud()
    • GradDistributionPlot.create_plot_dirs()
    • GradDistributionPlot.prepare_results_saver()
• model_load
  • ModelLoadContinueTraining
    • ModelLoadContinueTraining.on_train_loop_registration()
    • ModelLoadContinueTraining.on_train_begin()
    • ModelLoadContinueTraining.init_model_loader()
• model_save
  • ModelCheckpoint
    • ModelCheckpoint.on_epoch_end()
    • ModelCheckpoint.on_train_loop_registration()
    • ModelCheckpoint.save_hyperparams()
  • ModelIterationCheckpoint
    • ModelIterationCheckpoint.on_batch_end()
  • ModelTrainEndSave
    • ModelTrainEndSave.on_train_end()
    • ModelTrainEndSave.on_train_loop_registration()
    • ModelTrainEndSave.save_hyperparams()
    • ModelTrainEndSave.check_result_packages()
• performance_eval
  • ModelPerformanceEvaluation
    • ModelPerformanceEvaluation.on_train_end()
    • ModelPerformanceEvaluation.on_epoch_end()
    • ModelPerformanceEvaluation.evaluate_model_performance()
    • ModelPerformanceEvaluation.store_evaluated_metrics_to_history()
    • ModelPerformanceEvaluation.on_train_loop_registration()
  • ModelPerformancePrintReport
    • ModelPerformancePrintReport.on_train_end()
    • ModelPerformancePrintReport.on_epoch_end()
    • ModelPerformancePrintReport.print_performance_report()
  • TrainHistoryFormatter
    • TrainHistoryFormatter.on_epoch_end()
    • TrainHistoryFormatter.on_train_end()
    • TrainHistoryFormatter.format_history()
    • TrainHistoryFormatter.check_if_history_updated()
  • MetricHistoryRename
  • ModelTrainHistoryBaseCB
    • ModelTrainHistoryBaseCB.prepare_results_saver()
  • ModelTrainHistoryPlot
    • ModelTrainHistoryPlot.on_train_loop_registration()
    • ModelTrainHistoryPlot.on_epoch_end()
    • ModelTrainHistoryPlot.on_train_end()
    • ModelTrainHistoryPlot.plot_current_train_history()
  • ModelTrainHistoryFileWriter
    • ModelTrainHistoryFileWriter.on_train_loop_registration()
    • ModelTrainHistoryFileWriter.on_epoch_end()
    • ModelTrainHistoryFileWriter.on_train_end()
    • ModelTrainHistoryFileWriter.write_current_train_history()
• tensorboard
  • TensorboardReporterBaseCB
    • TensorboardReporterBaseCB.log_mid_train_loss()
    • TensorboardReporterBaseCB.log_train_history_metrics()
    • TensorboardReporterBaseCB.on_train_end()
    • TensorboardReporterBaseCB.on_train_loop_registration()
    • TensorboardReporterBaseCB.create_log_dir()
    • TensorboardReporterBaseCB.prepare_results_saver()
    • TensorboardReporterBaseCB.upload_to_cloud()
  • TensorboardTrainBatchLoss
    • TensorboardTrainBatchLoss.on_batch_end()
    • TensorboardTrainBatchLoss.on_epoch_end()
  • TensorboardTrainHistoryMetric
    • TensorboardTrainHistoryMetric.on_epoch_end()
  • TensorboardFullTracking
    • TensorboardFullTracking.on_batch_end()
    • TensorboardFullTracking.on_epoch_end()
• train_schedule
• wandb
  • AlertConfig
    • AlertConfig.metric_name
    • AlertConfig.threshold_value
    • AlertConfig.objective
    • AlertConfig.wandb_alert_level
  • WandBTracking
    • WandBTracking.on_epoch_end()
    • WandBTracking.on_batch_end()
    • WandBTracking.log_mid_train_loss()
    • WandBTracking.log_train_history_metrics()
    • WandBTracking.send_configured_alerts()
    • WandBTracking.on_train_loop_registration()
    • WandBTracking.try_infer_additional_logging_details()
    • WandBTracking.check_alerts()