sakura.utils
Miscellaneous Utilities
Submodules
Composable dataset splitting utilities |
|
Various data transformations utilities |
|
Batch sample generation functions of various distributions |
|
Gradient reversal layer for pytorch module |
|
KL divergence for loss computation |
|
Logger of SAKURA pipeline |
|
(to be implemented) |
|
PyTorch Model Memory Estimator |
|
Sliced Wasserstein distance for loss computation |