All modules for which code is available
- torch_uncertainty.baselines.classification.resnet
- torch_uncertainty.baselines.classification.vgg
- torch_uncertainty.baselines.classification.wideresnet
- torch_uncertainty.baselines.depth.bts
- torch_uncertainty.baselines.regression.mlp
- torch_uncertainty.baselines.segmentation.deeplab
- torch_uncertainty.baselines.segmentation.segformer
- torch_uncertainty.datamodules.classification.cifar10
- torch_uncertainty.datamodules.classification.cifar100
- torch_uncertainty.datamodules.classification.imagenet
- torch_uncertainty.datamodules.classification.mnist
- torch_uncertainty.datamodules.classification.tiny_imagenet
- torch_uncertainty.datamodules.segmentation.camvid
- torch_uncertainty.datamodules.segmentation.cityscapes
- torch_uncertainty.datamodules.segmentation.muad
- torch_uncertainty.datamodules.uci_regression
- torch_uncertainty.datasets.classification.cifar.cifar_c
- torch_uncertainty.datasets.classification.cifar.cifar_h
- torch_uncertainty.datasets.classification.cifar.cifar_n
- torch_uncertainty.datasets.classification.imagenet.imagenet_a
- torch_uncertainty.datasets.classification.imagenet.imagenet_c
- torch_uncertainty.datasets.classification.imagenet.imagenet_o
- torch_uncertainty.datasets.classification.imagenet.imagenet_r
- torch_uncertainty.datasets.classification.imagenet.tiny_imagenet
- torch_uncertainty.datasets.classification.imagenet.tiny_imagenet_c
- torch_uncertainty.datasets.classification.mnist_c
- torch_uncertainty.datasets.classification.not_mnist
- torch_uncertainty.datasets.classification.openimage_o
- torch_uncertainty.datasets.fractals
- torch_uncertainty.datasets.frost
- torch_uncertainty.datasets.kitti
- torch_uncertainty.datasets.muad
- torch_uncertainty.datasets.nyu
- torch_uncertainty.datasets.regression.uci_regression
- torch_uncertainty.datasets.segmentation.camvid
- torch_uncertainty.datasets.segmentation.cityscapes
- torch_uncertainty.layers.batch_ensemble
- torch_uncertainty.layers.bayesian.bayes_conv
- torch_uncertainty.layers.bayesian.bayes_linear
- torch_uncertainty.layers.bayesian.lpbnn
- torch_uncertainty.layers.masksembles
- torch_uncertainty.layers.packed
- torch_uncertainty.losses.bayesian
- torch_uncertainty.losses.classification
- torch_uncertainty.losses.regression
- torch_uncertainty.metrics.classification.adaptive_calibration_error
- torch_uncertainty.metrics.classification.brier_score
- torch_uncertainty.metrics.classification.calibration_error
- torch_uncertainty.metrics.classification.categorical_nll
- torch_uncertainty.metrics.classification.disagreement
- torch_uncertainty.metrics.classification.entropy
- torch_uncertainty.metrics.classification.fpr
- torch_uncertainty.metrics.classification.grouping_loss
- torch_uncertainty.metrics.classification.mean_iou
- torch_uncertainty.metrics.classification.mutual_information
- torch_uncertainty.metrics.classification.risk_coverage
- torch_uncertainty.metrics.classification.variation_ratio
- torch_uncertainty.metrics.regression.inverse
- torch_uncertainty.metrics.regression.log10
- torch_uncertainty.metrics.regression.mse_log
- torch_uncertainty.metrics.regression.nll
- torch_uncertainty.metrics.regression.relative_error
- torch_uncertainty.metrics.regression.silog
- torch_uncertainty.metrics.regression.threshold_accuracy
- torch_uncertainty.metrics.sparsification
- torch_uncertainty.models.wrappers.checkpoint_ensemble
- torch_uncertainty.models.wrappers.deep_ensembles
- torch_uncertainty.models.wrappers.ema
- torch_uncertainty.models.wrappers.mc_dropout
- torch_uncertainty.models.wrappers.stochastic
- torch_uncertainty.models.wrappers.swa
- torch_uncertainty.models.wrappers.swag
- torch_uncertainty.post_processing.calibration.matrix_scaler
- torch_uncertainty.post_processing.calibration.temperature_scaler
- torch_uncertainty.post_processing.calibration.vector_scaler
- torch_uncertainty.post_processing.laplace
- torch_uncertainty.post_processing.mc_batch_norm
- torch_uncertainty.routines.classification
- torch_uncertainty.routines.pixel_regression
- torch_uncertainty.routines.regression
- torch_uncertainty.routines.segmentation