sakura.utils

Miscellaneous Utilities

Submodules

sakura.utils.data_splitter

Composable dataset splitting utilities

sakura.utils.data_transformations

Various data transformations utilities

sakura.utils.distributions

Batch sample generation functions of various distributions

sakura.utils.gradient_reverse

Gradient reversal layer for pytorch module

sakura.utils.kl_divergence

KL divergence for loss computation

sakura.utils.logger

Logger of SAKURA pipeline

sakura.utils.par_perturber

(to be implemented)

sakura.utils.size_estimator

PyTorch Model Memory Estimator

sakura.utils.sliced_wasserstein

Sliced Wasserstein distance for loss computation