cyto_dl.callbacks package

Submodules

• cyto_dl.callbacks.callback_utils module
• cyto_dl.callbacks.grad_monitor module
  • GradientLoggingCallback
    • GradientLoggingCallback.on_train_epoch_end()
• cyto_dl.callbacks.image_saver module
  • ImageSaver
    • ImageSaver.on_predict_batch_end()
    • ImageSaver.on_test_batch_end()
    • ImageSaver.on_train_batch_end()
    • ImageSaver.on_validation_batch_end()
    • ImageSaver.save()
• cyto_dl.callbacks.latent_walk module
• cyto_dl.callbacks.latent_walk_diffae module
  • DiffAELatentWalk
    • DiffAELatentWalk.on_predict_epoch_end()
    • DiffAELatentWalk.on_validation_batch_end()
    • DiffAELatentWalk.on_validation_epoch_end()
• cyto_dl.callbacks.layer_freeze module
  • LayerFreeze
    • LayerFreeze.setup()
• cyto_dl.callbacks.model_utils module
  • GetEmbeddings
    • GetEmbeddings.on_test_epoch_end()
  • get_all_embeddings()
  • save_predictions_classifier()
• cyto_dl.callbacks.outlier_detection module
  • OutlierDetection
    • OutlierDetection.calculate_mahalanobis_hook()
    • OutlierDetection.flatten_activations()
    • OutlierDetection.on_load_checkpoint()
    • OutlierDetection.on_predict_batch_end()
    • OutlierDetection.on_predict_epoch_start()
    • OutlierDetection.on_save_checkpoint()
    • OutlierDetection.on_test_batch_end()
    • OutlierDetection.on_test_epoch_start()
    • OutlierDetection.on_train_epoch_start()
    • OutlierDetection.update_covariance_hook()
• cyto_dl.callbacks.tabular_saver module
  • SaveTabularData
    • SaveTabularData.on_predict_epoch_end()
    • SaveTabularData.pred_to_df()
    • SaveTabularData.save_feats()