Index

A

AbaloneUCIRegressionDataModule (class in yamle.data.regression) accelerator (yamle.trainers.trainer.BaseTrainer parameter) Activation (class in yamle.models.operations) activation (yamle.models.convnet.ConvNetModel parameter) (yamle.models.fc.FCModel parameter) (yamle.models.operations.Activation parameter) (yamle.models.operations.Conv2dExtractor parameter) (yamle.models.operations.Conv2dNormActivation parameter) (yamle.models.operations.DoubleConv2d parameter) (yamle.models.operations.LinearExtractor parameter) (yamle.models.operations.LinearNormActivation parameter) (yamle.models.operations.SqueezeAndExcitation parameter) activation_bits (yamle.quantization.quantizer.BaseQuantizer parameter) Add (class in yamle.models.operations) add_method_specific_layers() (yamle.models.convnet.ConvNetModel method) (yamle.models.densenet.DenseNetModel method) (yamle.models.fc.FCModel method) (yamle.models.gp.GPModel method) (yamle.models.mixer.MixerModel method) (yamle.models.model.BaseModel method) (yamle.models.resnet.ResNetModel method) (yamle.models.rnn.RNNModel method) (yamle.models.transformer.TransformerModel method) (yamle.models.unet.UNetModel method) (yamle.models.vgg.VGGModel method) (yamle.models.visual_transformer.VisualTransformerModel method) add_shared_args() (in module yamle.utils.cli_utils) add_specific_args() (yamle.data.classification.MedMNISTClassificationDataModule static method) (yamle.data.classification.TorchvisionClassificationDataModule static method) (yamle.data.classification.TorchvisionClassificationDataModuleCIFAR3 static method) (yamle.data.classification.TorchvisionClassificationDataModuleCIFAR5 static method) (yamle.data.classification.ToyTwoMoonsClassificationDataModule static method) (yamle.data.datamodule.BaseDataModule static method) (yamle.data.reconstruction.ECG5000ReconstructionDataModule static method) (yamle.data.regression.MedMNISTRegressionDataModule static method) (yamle.data.regression.TimeSeriesRegressionDataModule static method) (yamle.data.regression.TorchvisionRotationRegressionDataModule static method) (yamle.data.regression.ToyRegressionDataModule static method) (yamle.losses.classification.CrossEntropyLoss static method) (yamle.losses.contrastive.NoiseContrastiveEstimatorLoss static method) (yamle.losses.loss.BaseLoss static method) (yamle.losses.regression.GaussianNegativeLogLikelihoodLoss static method) (yamle.losses.regression.MeanSquaredError static method) (yamle.losses.regression.QuantileRegressionLoss static method) (yamle.losses.segmentation.FocalLoss static method) (yamle.losses.segmentation.SoftIntersectionOverUnionLoss static method) (yamle.methods.augmentation_classification.CutMixImageClassificationMethod static method) (yamle.methods.augmentation_classification.CutOutImageClassificationMethod static method) (yamle.methods.augmentation_classification.MixUpImageClassificationMethod static method) (yamle.methods.augmentation_classification.RandomErasingImageClassificationMethod static method) (yamle.methods.contrastive.SimCLRVisionMethod static method) (yamle.methods.delta_uq.DeltaUQMethod static method) (yamle.methods.dun.DUNMethod static method) (yamle.methods.early_exit.EarlyExitMethod static method) (yamle.methods.ensemble.GradientBoostingEnsembleMethod static method) (yamle.methods.evidential_regression.EvidentialRegressionMethod static method) (yamle.methods.gp.GPMethod static method) (yamle.methods.laplace.LaplaceMethod static method) (yamle.methods.mcdropout.MCDropBlockMethod static method) (yamle.methods.mcdropout.MCDropConnectMethod static method) (yamle.methods.mcdropout.MCDropoutMethod static method) (yamle.methods.mcdropout.MCStandOutMethod static method) (yamle.methods.mcdropout.MCStochasticDepthMethod static method) (yamle.methods.method.BaseMethod static method) (yamle.methods.mimo.DataMUXMethod static method) (yamle.methods.mimo.MIMMOMethod static method) (yamle.methods.mimo.MIMOMethod static method) (yamle.methods.mimo.MixMoMethod static method) (yamle.methods.mimo.MixVitMethod static method) (yamle.methods.mimo.UnMixMoMethod static method) (yamle.methods.moe.MixtureOfExpertsMethod static method) (yamle.methods.moe.MultiHeadEnsembleMethod static method) (yamle.methods.pe.PEMethod static method) (yamle.methods.rbnn.RBNNMethod static method) (yamle.methods.sgld.SGLDMethod static method) (yamle.methods.sngp.SNGPMethod static method) (yamle.methods.svi.SVIReparameterizationMethod static method) (yamle.methods.swag.SWAGMethod static method) (yamle.methods.temperature_scaling.TemperatureMethod static method) (yamle.methods.uncertain_method.MemberMethod static method) (yamle.methods.uncertain_method.SVIMethod static method) (yamle.models.convnet.ConvNetModel static method) (yamle.models.densenet.DenseNetModel static method) (yamle.models.fc.FCModel static method) (yamle.models.gp.GPModel static method) (yamle.models.mixer.MixerModel static method) (yamle.models.model.BaseModel class method) (yamle.models.resnet.ResNetModel static method) (yamle.models.rnn.RNNAutoEncoderModel static method) (yamle.models.rnn.RNNModel static method) (yamle.models.transformer.TransformerModel static method) (yamle.models.unet.UNetModel static method) (yamle.models.vgg.VGGModel static method) (yamle.models.visual_transformer.VisualTransformerModel static method) (yamle.pruning.pruner.BasePruner static method) (yamle.pruning.unstructured.magnitude.UnstructuredMagnitudePruner static method) (yamle.quantization.qat.QATQuantizer static method) (yamle.quantization.quantizer.BaseQuantizer static method) (yamle.regularizers.feature.InnerProductFeatureRegularizer static method) (yamle.regularizers.gradient.GradientNoiseRegularizer static method) (yamle.regularizers.model.ShrinkAndPerturbRegularizer static method) (yamle.regularizers.regularizer.BaseRegularizer static method) (yamle.trainers.calibration.CalibrationTrainer static method) (yamle.trainers.ensemble.EnsembleTrainer static method) (yamle.trainers.trainer.BaseTrainer static method) add_test_args() (in module yamle.utils.cli_utils) add_train_args() (in module yamle.utils.cli_utils) additional_heads (yamle.methods.mimo.MIMMOMethod parameter) additive_gaussian_noise() (in module yamle.third_party.imagenet_c.tabular_corruptions) additive_uniform_noise() (in module yamle.third_party.imagenet_c.tabular_corruptions) AdultIncomeUCIClassificationDataModule (class in yamle.data.classification) alignment (yamle.models.operations.Reduction parameter) allow_zero_length_dataloader_with_multiple_devices (yamle.methods.augmentation_classification.CutMixImageClassificationMethod attribute) (yamle.methods.augmentation_classification.CutOutImageClassificationMethod attribute) (yamle.methods.augmentation_classification.MixUpImageClassificationMethod attribute) (yamle.methods.augmentation_classification.RandomErasingImageClassificationMethod attribute) (yamle.methods.mimo.DataMUXMethod attribute) (yamle.methods.mimo.MIMMOMethod attribute) (yamle.methods.mimo.MIMOMethod attribute) (yamle.methods.mimo.MixMoMethod attribute) (yamle.methods.mimo.MixVitMethod attribute) (yamle.methods.mimo.UnMixMoMethod attribute) (yamle.methods.moe.MixtureOfExpertsMethod attribute) (yamle.methods.moe.MultiHeadEnsembleMethod attribute) (yamle.methods.pe.PEMethod attribute) (yamle.methods.rbnn.RBNNMethod attribute) (yamle.methods.sgld.SGLDMethod attribute) (yamle.methods.sngp.SNGPMethod attribute) (yamle.methods.svi.SVIFlipOutDropConnectMethod attribute) (yamle.methods.svi.SVIFlipOutRTMethod attribute) (yamle.methods.svi.SVILRTMethod attribute) (yamle.methods.svi.SVILRTVDMethod attribute) (yamle.methods.svi.SVIReparameterizationMethod attribute) (yamle.methods.svi.SVIRTMethod attribute) (yamle.methods.swag.SWAGMethod attribute) (yamle.methods.temperature_scaling.TemperatureMethod attribute) (yamle.methods.uncertain_method.MCSamplingMethod attribute) (yamle.methods.uncertain_method.MemberMethod attribute) (yamle.methods.uncertain_method.SVIMethod attribute) alpha (yamle.data.augmentations.CutMix parameter) (yamle.data.augmentations.MixUp parameter) (yamle.losses.segmentation.FocalLoss parameter) (yamle.methods.dun.DUNMethod parameter) (yamle.methods.early_exit.EarlyExitMethod parameter) (yamle.methods.evidential_regression.EvidentialRegressionMethod parameter) (yamle.methods.mcdropout.MCStandOutMethod parameter) (yamle.methods.mimo.MIMMOMethod parameter) (yamle.methods.mimo.MixMoMethod parameter) (yamle.methods.moe.MixtureOfExpertsMethod parameter) (yamle.methods.pe.PEMethod parameter) (yamle.methods.pe.replace_layers_with_grouped_convs parameter) (yamle.methods.uncertain_method.SVIMethod parameter)

alphas (yamle.methods.dun.DUNMethod property) analyse() (yamle.methods.method.BaseMethod method) (yamle.methods.pe.PEMethod method) anomaly (yamle.data.reconstruction.ECG5000ReconstructionDataModule parameter) apply_to_normalisation (yamle.methods.swag.SWAGMethod parameter) argparse_to_command() (in module yamle.utils.file_utils) argparse_to_config_file() (in module yamle.utils.file_utils) argparse_to_dictionary() (in module yamle.utils.file_utils) args (yamle.utils.file_utils.argparse_to_config_file parameter) (yamle.utils.file_utils.save_experiment parameter) (yamle.utils.running_utils.extract_kwargs parameter) args_dictionary_file() (in module yamle.utils.file_utils) args_file() (in module yamle.utils.file_utils) Attention (class in yamle.models.transformer) augment() (yamle.data.augmentations.Augmentation method) (yamle.data.augmentations.CutMix method) (yamle.data.augmentations.CutOut method) (yamle.data.augmentations.MixUp method) (yamle.data.augmentations.RandomErasing method) Augmentation (class in yamle.data.augmentations) augmentation (yamle.data.custom.ECG5000DataModule.setup parameter) (yamle.data.datamodule.RealWorldDataModule.setup parameter) (yamle.data.datamodule.VisionDataModule.setup parameter) AugmentationImageClassificationMethod (class in yamle.methods.augmentation_classification) available_augmentations (yamle.third_party.imagenet_c.TabularCorruption attribute) (yamle.third_party.imagenet_c.VisionCorruption attribute) available_heads (yamle.methods.mimo.MIMMOMethod parameter) available_test_augmentations (yamle.data.classification.BloodMNISTClassificationDataModule attribute) (yamle.data.classification.BreastMNISTClassificationDataModule attribute) (yamle.data.classification.DermaMNISTClassificationDataModule attribute) (yamle.data.classification.PneumoniaMNISTClassificationDataModule attribute) (yamle.data.classification.TinyImageNetClassificationDataModule attribute) (yamle.data.classification.TorchvisionClassificationDataModuleCIFAR10 attribute) (yamle.data.classification.TorchvisionClassificationDataModuleCIFAR100 attribute) (yamle.data.classification.TorchvisionClassificationDataModuleCIFAR3 attribute) (yamle.data.classification.TorchvisionClassificationDataModuleCIFAR5 attribute) (yamle.data.classification.TorchvisionClassificationDataModuleFashionMNIST attribute) (yamle.data.classification.TorchvisionClassificationDataModuleMNIST attribute) (yamle.data.classification.TorchvisionClassificationDataModuleSVHN attribute) (yamle.data.classification.ToyTwoCirclesClassificationDataModule attribute) (yamle.data.datamodule.RealWorldClassificationDataModule attribute) (yamle.data.datamodule.RealWorldDataModule attribute) (yamle.data.datamodule.RealWorldRegressionDataModule attribute) (yamle.data.datamodule.VisionClassificationDataModule attribute) (yamle.data.datamodule.VisionRegressionDataModule attribute) (yamle.data.depth.DepthEstimationDataModule attribute) (yamle.data.depth.NYUv2DataModule attribute) (yamle.data.reconstruction.ECG5000ReconstructionDataModule attribute) (yamle.data.regression.AbaloneUCIRegressionDataModule attribute) (yamle.data.regression.BostonUCIRegressionDataModule attribute) (yamle.data.regression.ConcreteUCIRegressionDataModule attribute) (yamle.data.regression.EnergyUCIRegressionDataModule attribute) (yamle.data.regression.MedMNISTRegressionDataModule attribute) (yamle.data.regression.RetinaMNISTDataModule attribute) (yamle.data.regression.TelemonitoringUCIRegressionDataModule attribute) (yamle.data.regression.TemperatureTimeSeriesDataModule attribute) (yamle.data.regression.TimeSeriesRegressionDataModule attribute) (yamle.data.regression.TinyImageNetRotationRegressionDataModule attribute) (yamle.data.regression.TorchvisionRegressionDataModule attribute) (yamle.data.regression.TorchvisionRotationRegressionDataModule attribute) (yamle.data.regression.TorchvisionRotationRegressionDataModuleCIFAR10 attribute) (yamle.data.regression.TorchvisionRotationRegressionDataModuleCIFAR100 attribute) (yamle.data.regression.TorchvisionRotationRegressionDataModuleFashionMNIST attribute) (yamle.data.regression.TorchvisionRotationRegressionDataModuleMNIST attribute) (yamle.data.regression.TorchvisionRotationRegressionDataModuleSVHN attribute) (yamle.data.regression.ToyRegressionDataModule attribute) (yamle.data.regression.UCIRegressionDataModule attribute) (yamle.data.regression.WikiFaceRegressionDataModule attribute) (yamle.data.regression.WineQualityUCIRegressionDataModule attribute) (yamle.data.regression.YachtUCIRegressionDataModule attribute) (yamle.data.segmentation.TorchvisionSegmentationDataModule attribute) (yamle.data.segmentation.TorchvisionSegmentationDataModuleCityscapes attribute) (yamle.data.text.Shakespeare attribute) (yamle.data.text.TorchtextClassificationDataModule attribute) (yamle.data.text.TorchtextClassificationModelIMDB attribute) (yamle.data.text.TorchtextClassificationModelWikiText103 attribute) (yamle.data.text.TorchtextClassificationModelWikiText2 attribute) available_transforms (yamle.data.classification.BloodMNISTClassificationDataModule attribute) (yamle.data.classification.BreastMNISTClassificationDataModule attribute) (yamle.data.classification.DermaMNISTClassificationDataModule attribute) (yamle.data.classification.PneumoniaMNISTClassificationDataModule attribute) (yamle.data.classification.TinyImageNetClassificationDataModule attribute) (yamle.data.classification.TorchvisionClassificationDataModuleCIFAR10 attribute) (yamle.data.classification.TorchvisionClassificationDataModuleCIFAR100 attribute) (yamle.data.classification.TorchvisionClassificationDataModuleCIFAR3 attribute) (yamle.data.classification.TorchvisionClassificationDataModuleCIFAR5 attribute) (yamle.data.classification.TorchvisionClassificationDataModuleFashionMNIST attribute) (yamle.data.classification.TorchvisionClassificationDataModuleMNIST attribute) (yamle.data.classification.TorchvisionClassificationDataModuleSVHN attribute) (yamle.data.classification.ToyTwoCirclesClassificationDataModule attribute) (yamle.data.datamodule.RealWorldClassificationDataModule attribute) (yamle.data.datamodule.RealWorldDataModule attribute) (yamle.data.datamodule.RealWorldRegressionDataModule attribute) (yamle.data.datamodule.VisionClassificationDataModule attribute) (yamle.data.datamodule.VisionRegressionDataModule attribute) (yamle.data.depth.DepthEstimationDataModule attribute) (yamle.data.depth.NYUv2DataModule attribute) (yamle.data.reconstruction.ECG5000ReconstructionDataModule attribute) (yamle.data.regression.AbaloneUCIRegressionDataModule attribute) (yamle.data.regression.BostonUCIRegressionDataModule attribute) (yamle.data.regression.ConcreteUCIRegressionDataModule attribute) (yamle.data.regression.EnergyUCIRegressionDataModule attribute) (yamle.data.regression.MedMNISTRegressionDataModule attribute) (yamle.data.regression.RetinaMNISTDataModule attribute) (yamle.data.regression.TelemonitoringUCIRegressionDataModule attribute) (yamle.data.regression.TemperatureTimeSeriesDataModule attribute) (yamle.data.regression.TimeSeriesRegressionDataModule attribute) (yamle.data.regression.TinyImageNetRotationRegressionDataModule attribute) (yamle.data.regression.TorchvisionRegressionDataModule attribute) (yamle.data.regression.TorchvisionRotationRegressionDataModule attribute) (yamle.data.regression.TorchvisionRotationRegressionDataModuleCIFAR10 attribute) (yamle.data.regression.TorchvisionRotationRegressionDataModuleCIFAR100 attribute) (yamle.data.regression.TorchvisionRotationRegressionDataModuleFashionMNIST attribute) (yamle.data.regression.TorchvisionRotationRegressionDataModuleMNIST attribute) (yamle.data.regression.TorchvisionRotationRegressionDataModuleSVHN attribute) (yamle.data.regression.ToyRegressionDataModule attribute) (yamle.data.regression.UCIRegressionDataModule attribute) (yamle.data.regression.WikiFaceRegressionDataModule attribute) (yamle.data.regression.WineQualityUCIRegressionDataModule attribute) (yamle.data.regression.YachtUCIRegressionDataModule attribute) (yamle.data.segmentation.TorchvisionSegmentationDataModule attribute) (yamle.data.segmentation.TorchvisionSegmentationDataModuleCityscapes attribute) (yamle.data.text.Shakespeare attribute) (yamle.data.text.TorchtextClassificationDataModule attribute) (yamle.data.text.TorchtextClassificationModelIMDB attribute) (yamle.data.text.TorchtextClassificationModelWikiText103 attribute) (yamle.data.text.TorchtextClassificationModelWikiText2 attribute) average_augmentations() (in module yamle.utils.plotting_utils) average_predictions() (in module yamle.utils.operation_utils)

B

backward() (yamle.methods.method.BaseMethod method) BaggingTrainer (class in yamle.trainers.ensemble) base_folder (yamle.third_party.tinyimagenet.TinyImageNet attribute) base_width (yamle.models.resnet.Bottleneck attribute) base_width_multiplier (yamle.models.resnet.Bottleneck attribute) BaseDataModule (class in yamle.data.datamodule) BaseLoss (class in yamle.losses.loss) BaseMethod (class in yamle.methods.method) BaseModel (class in yamle.models.model) BasePruner (class in yamle.pruning.pruner) BaseQuantizer (class in yamle.quantization.quantizer) BaseRegularizer (class in yamle.regularizers.regularizer) BaseTrainer (class in yamle.trainers.trainer) BasicBlock (class in yamle.models.resnet) batch_proportion (yamle.data.augmentations.Augmentation parameter) (yamle.data.augmentations.CutMix parameter) (yamle.data.augmentations.CutOut parameter) (yamle.data.augmentations.RandomErasing parameter) (yamle.methods.augmentation_classification.CutMixImageClassificationMethod parameter) (yamle.methods.augmentation_classification.CutOutImageClassificationMethod parameter) (yamle.methods.augmentation_classification.MixUpImageClassificationMethod parameter) (yamle.methods.augmentation_classification.RandomErasingImageClassificationMethod parameter) batch_size (yamle.data.datamodule.BaseDataModule parameter) BEMethod (class in yamle.methods.be) best_config_to_command_arguments() (in module yamle.utils.tuning_utils)

beta (yamle.methods.early_exit.EarlyExitMethod parameter) (yamle.methods.mcdropout.MCStandOutMethod parameter) (yamle.methods.moe.MixtureOfExpertsMethod parameter) bias (yamle.models.operations.CompletelySeparableConv2d parameter) (yamle.models.operations.Conv2dNormActivation parameter) (yamle.models.operations.DepthwiseSeparableConv2d parameter) (yamle.models.operations.DoubleConv2d parameter) (yamle.models.operations.LinearNormActivation parameter) block (yamle.models.resnet.Blocks parameter) (yamle.models.resnet.ResNetModel parameter) block_size_percentage (yamle.methods.mcdropout.MCDropBlockMethod parameter) Blocks (class in yamle.models.resnet) (class in yamle.models.vgg) blocks (yamle.models.resnet.Blocks parameter) (yamle.models.vgg.Blocks parameter) BloodMNISTClassificationDataModule (class in yamle.data.classification) bn_function() (yamle.models.densenet.DenseLayer method) bn_size (yamle.models.densenet.DenseBlock parameter) (yamle.models.densenet.DenseLayer parameter) (yamle.models.densenet.DenseNetModel parameter) BostonUCIRegressionDataModule (class in yamle.data.regression) Bottleneck (class in yamle.models.resnet) BreastCancerUCIClassificationDataModule (class in yamle.data.classification) BreastMNISTClassificationDataModule (class in yamle.data.classification) brightness() (in module yamle.third_party.imagenet_c.corruptions)

C

c (yamle.models.operations.LSTM.forward parameter) calculate_overlap_between_norms() (in module yamle.utils.specific.mimo_experiments.plotting_utils) calibrate() (yamle.methods.temperature_scaling.TemperatureMethod method) (yamle.trainers.trainer.BaseTrainer method) calibration_dataloader() (yamle.data.datamodule.BaseDataModule method) calibration_dataset() (yamle.data.datamodule.BaseDataModule method) calibration_dataset_size() (yamle.data.datamodule.BaseDataModule method) calibration_epochs (yamle.trainers.calibration.CalibrationTrainer parameter) calibration_joint_transform() (yamle.data.datamodule.BaseDataModule method) calibration_learning_rate (yamle.methods.temperature_scaling.TemperatureMethod parameter) calibration_number_of_batches() (yamle.data.datamodule.BaseDataModule method) calibration_portion (yamle.data.datamodule.BaseDataModule parameter) calibration_target_transform() (yamle.data.datamodule.BaseDataModule method) calibration_transform() (yamle.data.datamodule.BaseDataModule method) CalibrationTrainer (class in yamle.trainers.calibration) CarEvaluationUCIClassificationDataModule (class in yamle.data.classification) causal (yamle.models.transformer.Attention parameter) (yamle.models.transformer.TransformerEncoderLayer parameter) channels_dim (yamle.models.mixer.MixerLayer parameter) channels_hidden_dim (yamle.models.mixer.MixerLayer parameter) (yamle.models.mixer.MixerModel parameter) class_weights (yamle.losses.classification.CrossEntropyLoss parameter) classification_uncertainty_decomposition() (in module yamle.utils.operation_utils) ClassificationDatasetSubset (class in yamle.data.transforms) cleanup() (yamle.quantization.qat.QATQuantizer method) (yamle.quantization.quantizer.BaseQuantizer method) clipped_zoom() (in module yamle.third_party.imagenet_c.corruptions) coder_depth (yamle.methods.mimo.DataMUXMethod parameter) coder_expansion_factor (yamle.methods.mimo.DataMUXMethod parameter) coeff (yamle.methods.sngp.SNGPMethod parameter) compile (yamle.trainers.trainer.BaseTrainer parameter) complete (yamle.methods.method.BaseMethod.reset_metrics parameter) CompletelySeparableConv2d (class in yamle.models.operations) ConcreteUCIRegressionDataModule (class in yamle.data.regression)

config_file (yamle.utils.file_utils.argparse_to_config_file parameter) config_file() (in module yamle.utils.file_utils) config_logger() (in module yamle.utils.file_utils) configure_optimizers() (yamle.methods.ensemble.SnapsotEnsembleMethod method) (yamle.methods.method.BaseMethod method) (yamle.methods.temperature_scaling.TemperatureMethod method) contrast() (in module yamle.third_party.imagenet_c.corruptions) Conv2dExtractor (class in yamle.models.operations) Conv2dNormActivation (class in yamle.models.operations) conv_filter_dropout (yamle.methods.mcdropout.MCDropoutMethod parameter) conv_hidden_dims (yamle.models.convnet.ConvNetModel parameter) ConvNetModel (class in yamle.models.convnet) convolution (yamle.models.operations.Conv2dExtractor parameter) CorrelationFeatureRegularizer (class in yamle.regularizers.feature) corruption_dict (yamle.third_party.imagenet_c.TabularCorruption attribute) (yamle.third_party.imagenet_c.VisionCorruption attribute) corruption_name (yamle.third_party.imagenet_c.TabularCorruption parameter) (yamle.third_party.imagenet_c.VisionCorruption parameter) corruption_names (yamle.third_party.imagenet_c.TabularCorruption attribute) (yamle.third_party.imagenet_c.VisionCorruption attribute) corruption_tuple (yamle.third_party.imagenet_c.TabularCorruption attribute) (yamle.third_party.imagenet_c.VisionCorruption attribute) CosineScalarScheduler (class in yamle.utils.optimization_utils) CosineSimilarityFeatureRegularizer (class in yamle.regularizers.feature) covariance (yamle.methods.swag.SWAGMethod parameter) create_experiment_folder() (in module yamle.utils.file_utils) CreditUCIClassificationDataModule (class in yamle.data.classification) CrossEntropyLoss (class in yamle.losses.classification) current_time() (in module yamle.utils.file_utils) CutMix (class in yamle.data.augmentations) cutmix_alpha (yamle.methods.augmentation_classification.CutMixImageClassificationMethod parameter) CutMixImageClassificationMethod (class in yamle.methods.augmentation_classification) CutOut (class in yamle.data.augmentations) cutout_size (yamle.methods.augmentation_classification.CutOutImageClassificationMethod parameter) CutOutImageClassificationMethod (class in yamle.methods.augmentation_classification)

D

data (yamle.data.transforms.Normalize.inverse parameter) data_dir (yamle.data.datamodule.BaseDataModule parameter) (yamle.data.datamodule.VisionDataModule parameter) (yamle.data.segmentation.TorchvisionSegmentationDataModule parameter) (yamle.data.text.TorchtextClassificationDataModule parameter) data_factory() (in module yamle.data) data_shape (yamle.data.datamodule.BaseDataModule attribute) datamodule (yamle.methods.method.BaseMethod parameter) (yamle.trainers.trainer.BaseTrainer parameter) DataMUXMethod (class in yamle.methods.mimo) dataset (yamle.data.classification.MedMNISTClassificationDataModule parameter) (yamle.data.classification.TorchvisionClassificationDataModule parameter) (yamle.data.classification.UCIClassificationDataModule parameter) (yamle.data.dataset_wrappers.ImageRotationDataset parameter) (yamle.data.dataset_wrappers.InputImagePaddingDataset parameter) (yamle.data.dataset_wrappers.SurrogateDataset parameter) (yamle.data.depth.DepthEstimationDataModule parameter) (yamle.data.regression.MedMNISTRegressionDataModule parameter) (yamle.data.regression.TorchvisionRotationRegressionDataModule parameter) (yamle.data.regression.UCIRegressionDataModule parameter) (yamle.data.segmentation.TorchvisionSegmentationDataModule parameter) (yamle.data.text.TorchtextClassificationDataModule parameter) (yamle.data.transforms.ClassificationDatasetSubset parameter) (yamle.third_party.medmnist.MedMNISTDatasetWrapper parameter) dataset_folder (yamle.third_party.tinyimagenet.TinyImageNet property) debug (yamle.trainers.trainer.BaseTrainer parameter) decoder_depth (yamle.models.rnn.RNNAutoEncoderModel parameter) defocus_blur() (in module yamle.third_party.imagenet_c.corruptions) DeltaUQMethod (class in yamle.methods.delta_uq) Denormalize (class in yamle.data.transforms) dense (yamle.models.transformer.FeedForward parameter) DenseBlock (class in yamle.models.densenet) DenseLayer (class in yamle.models.densenet) DenseNetModel (class in yamle.models.densenet) depth (yamle.methods.mimo.MixVitMethod parameter) (yamle.models.densenet.DenseNetModel parameter) (yamle.models.fc.FCModel parameter) (yamle.models.operations.Conv2dExtractor parameter) (yamle.models.operations.LinearExtractor parameter) (yamle.models.resnet.ResNetModel parameter) (yamle.models.rnn.RNNModel parameter) (yamle.models.vgg.VGGModel parameter) (yamle.models.visual_transformer.VisualTransformerModel parameter) depth_indices (yamle.methods.mcdropout.MCDropoutMethod parameter) depth_portion_to_replace_end (yamle.methods.mcdropout.MCDropBlockMethod parameter) (yamle.methods.mcdropout.MCDropConnectMethod parameter) (yamle.methods.mcdropout.MCDropoutMethod parameter) (yamle.methods.mcdropout.MCStandOutMethod parameter) depth_portion_to_replace_start (yamle.methods.mcdropout.MCDropBlockMethod parameter) (yamle.methods.mcdropout.MCDropConnectMethod parameter) (yamle.methods.mcdropout.MCDropoutMethod parameter) (yamle.methods.mcdropout.MCStandOutMethod parameter)

DepthEstimationDataModule (class in yamle.data.depth) DepthwiseSeparableConv2d (class in yamle.models.operations) DermaMNISTClassificationDataModule (class in yamle.data.classification) DermatologyUCIClassificationDataModule (class in yamle.data.classification) devices (yamle.trainers.trainer.BaseTrainer parameter) dilation (yamle.models.operations.Conv2dNormActivation parameter) dim (yamle.data.transforms.JointTargetSqueeze parameter) (yamle.models.operations.output_activation parameter) (yamle.models.operations.OutputActivation parameter) (yamle.models.operations.Reduction parameter) (yamle.models.transformer.Attention parameter) (yamle.models.transformer.FeedForward parameter) (yamle.models.transformer.PreNorm parameter) (yamle.models.transformer.TransformerEncoderLayer parameter) (yamle.regularizers.feature.InnerProductFeatureRegularizer parameter) (yamle.utils.operation_utils.entropy parameter) dim_head (yamle.models.transformer.Attention parameter) (yamle.models.transformer.TransformerEncoderLayer parameter) dimension (yamle.models.operations.Activation parameter) (yamle.models.operations.Normalization parameter) (yamle.models.operations.Pooling parameter) disable() (yamle.models.operations.OutputActivation method) (yamle.models.operations.ScalarMultiplier method) disable_optimization() (in module yamle.utils.optimization_utils) disable_pruning() (in module yamle.utils.pruning_utils) disable_regularizer() (in module yamle.utils.regularizer_utils) disk() (in module yamle.third_party.imagenet_c.corruptions) do_not_optimize_depth_weights (yamle.methods.mimo.MIMMOMethod parameter) DoubleConv2d (class in yamle.models.operations) DownBlock (class in yamle.models.unet) download (yamle.third_party.nyuv2.NYUv2 parameter) download() (yamle.third_party.nyuv2.NYUv2 method) (yamle.third_party.tinyimagenet.TinyImageNet method) download_depth() (in module yamle.third_party.nyuv2) download_rgb() (in module yamle.third_party.nyuv2) download_seg() (in module yamle.third_party.nyuv2) dropout (yamle.models.mixer.MixerLayer parameter) (yamle.models.mixer.MixerModel parameter) (yamle.models.transformer.Attention parameter) (yamle.models.transformer.FeedForward parameter) (yamle.models.transformer.PositionalEncoding parameter) (yamle.models.transformer.TransformerEncoderLayer parameter) (yamle.models.transformer.TransformerModel parameter) (yamle.models.visual_transformer.SpatialPositionalEmbedding parameter) (yamle.models.visual_transformer.VisualTransformerModel parameter) dropout_rate (yamle.models.densenet.DenseBlock parameter) (yamle.models.densenet.DenseLayer parameter) (yamle.models.densenet.DenseNetModel parameter) DummyLoss (class in yamle.losses.loss) DummyPruner (class in yamle.pruning.pruner) DummyQuantizer (class in yamle.quantization.quantizer) DummyRegularizer (class in yamle.regularizers.regularizer) DUNMethod (class in yamle.methods.dun)

E

EarlyExitMethod (class in yamle.methods.early_exit) ECG5000ClassificationDataModule (class in yamle.data.classification) ECG5000DataModule (class in yamle.data.custom) ECG5000ReconstructionDataModule (class in yamle.data.reconstruction) elastic_transform() (in module yamle.third_party.imagenet_c.corruptions) embedding_dim (yamle.models.transformer.PositionalEncoding parameter) (yamle.models.transformer.TransformerModel parameter) (yamle.models.visual_transformer.SpatialPositionalEmbedding parameter) (yamle.models.visual_transformer.VisualTransformerModel parameter) EmptyBlock (class in yamle.models.densenet) (class in yamle.models.vgg) enable() (yamle.models.operations.OutputActivation method) (yamle.models.operations.ScalarMultiplier method) enable_optimization() (in module yamle.utils.optimization_utils) enable_pruning() (in module yamle.utils.pruning_utils) enable_regularizer() (in module yamle.utils.regularizer_utils) enable_spectral_normalization() (in module yamle.methods.sngp) encoder_depth (yamle.models.rnn.RNNAutoEncoderModel parameter) end_activation (yamle.models.operations.Conv2dExtractor parameter) (yamle.models.operations.LinearExtractor parameter) end_epoch (yamle.regularizers.model.ShrinkAndPerturbRegularizer parameter) (yamle.utils.optimization_utils.CosineScalarScheduler parameter) (yamle.utils.optimization_utils.LinearScalarScheduler parameter) (yamle.utils.optimization_utils.PowerGrowthScalarScheduler parameter) (yamle.utils.optimization_utils.SineScalarScheduler parameter) end_normalization (yamle.models.operations.Conv2dExtractor parameter) (yamle.models.operations.LinearExtractor parameter) end_pooling (yamle.models.operations.Conv2dExtractor parameter) end_value (yamle.utils.optimization_utils.CosineScalarScheduler parameter) (yamle.utils.optimization_utils.LinearScalarScheduler parameter) (yamle.utils.optimization_utils.PowerGrowthScalarScheduler parameter) (yamle.utils.optimization_utils.SineScalarScheduler parameter)

EnergyUCIRegressionDataModule (class in yamle.data.regression) EnsembleMethod (class in yamle.methods.ensemble) EnsembleTrainer (class in yamle.trainers.ensemble) entropy() (in module yamle.utils.operation_utils) epoch_frequency (yamle.regularizers.model.ShrinkAndPerturbRegularizer parameter) epoch_offset (yamle.utils.trainer_utils.ValidationReporterMonitorCallback parameter) epochs (yamle.quantization.qat.QATQuantizer parameter) (yamle.trainers.trainer.BaseTrainer parameter) epochs_to_collect (yamle.methods.swag.SWAGMethod parameter) eta (yamle.regularizers.gradient.GradientNoiseRegularizer parameter) evaluation (yamle.methods.method.BaseMethod property) EvidentialRegressionLoss (class in yamle.losses.evidential_regression) EvidentialRegressionMethod (class in yamle.methods.evidential_regression) expansion (yamle.models.resnet.BasicBlock attribute) (yamle.models.resnet.Bottleneck attribute) expansion_factor (yamle.models.operations.Conv2dExtractor parameter) (yamle.models.operations.LinearExtractor parameter) export_onnx() (in module yamle.utils.export_utils) extra_repr() (yamle.models.mixer.MixerLayer method) (yamle.models.operations.Conv2dExtractor method) (yamle.models.operations.LinearExtractor method) (yamle.models.operations.OutputActivation method) (yamle.models.operations.Reduction method) (yamle.models.operations.ReshapeOutput method) (yamle.models.operations.ScalarMultiplier method) (yamle.models.transformer.Attention method) (yamle.models.transformer.FeedForward method) (yamle.models.transformer.TransformerEncoderLayer method) (yamle.third_party.tinyimagenet.TinyImageNet method) extract_kwargs() (in module yamle.utils.running_utils)

F

factor (yamle.losses.segmentation.SoftIntersectionOverUnionLoss parameter) FCModel (class in yamle.models.fc) FeedForward (class in yamle.models.transformer) fetch_value_and_std_from_result() (in module yamle.utils.plotting_utils) filename (yamle.third_party.tinyimagenet.TinyImageNet attribute) final_layer() (yamle.models.convnet.ConvNetModel method) (yamle.models.densenet.DenseNetModel method) (yamle.models.fc.FCModel method) (yamle.models.gp.GPModel method) (yamle.models.mixer.MixerModel method) (yamle.models.model.BaseModel method) (yamle.models.resnet.ResNetModel method) (yamle.models.rnn.RNNAutoEncoderModel method) (yamle.models.rnn.RNNModel method) (yamle.models.transformer.TransformerModel method) (yamle.models.unet.UNetModel method) (yamle.models.vgg.VGGModel method) (yamle.models.visual_transformer.VisualTransformerModel method) fine_tune() (yamle.trainers.trainer.BaseTrainer method) first (yamle.models.resnet.BasicBlock parameter) (yamle.models.resnet.Bottleneck parameter) fit() (yamle.trainers.calibration.CalibrationTrainer method) (yamle.trainers.ensemble.BaggingTrainer method) (yamle.trainers.ensemble.EnsembleTrainer method) (yamle.trainers.trainer.BaseTrainer method) flatten (yamle.losses.classification.CrossEntropyLoss parameter) (yamle.losses.regression.GaussianNegativeLogLikelihoodLoss parameter) (yamle.losses.regression.MeanSquaredError parameter) fn (yamle.models.operations.Lambda parameter) FocalLoss (class in yamle.losses.segmentation) fog() (in module yamle.third_party.imagenet_c.corruptions) forward() (yamle.losses.evidential_regression.NIG_NLL method) (yamle.losses.evidential_regression.NIG_Reg method) (yamle.models.convnet.ConvNetModel method) (yamle.models.densenet.DenseBlock method) (yamle.models.densenet.DenseLayer method) (yamle.models.densenet.DenseNetModel method) (yamle.models.densenet.EmptyBlock method) (yamle.models.densenet.Transition method) (yamle.models.fc.FCModel method) (yamle.models.gp.GPModel method) (yamle.models.mixer.MixerModel method) (yamle.models.model.BaseModel method) (yamle.models.operations.Activation method) (yamle.models.operations.Add method) (yamle.models.operations.CompletelySeparableConv2d method) (yamle.models.operations.Conv2dExtractor method) (yamle.models.operations.Conv2dNormActivation method) (yamle.models.operations.DepthwiseSeparableConv2d method) (yamle.models.operations.DoubleConv2d method) (yamle.models.operations.Lambda method) (yamle.models.operations.LinearExtractor method) (yamle.models.operations.LinearNormActivation method) (yamle.models.operations.LSTM method) (yamle.models.operations.MatrixMultiplication method) (yamle.models.operations.Multiply method) (yamle.models.operations.Normalization method) (yamle.models.operations.OutputActivation method) (yamle.models.operations.ParallelModel method) (yamle.models.operations.Pooling method) (yamle.models.operations.Reduction method) (yamle.models.operations.ReshapeInput method) (yamle.models.operations.ReshapeOutput method) (yamle.models.operations.ResidualLayer method) (yamle.models.operations.ScalarAdder method) (yamle.models.operations.ScalarMultiplier method) (yamle.models.operations.SqueezeAndExcitation method) (yamle.models.operations.Unsqueeze method) (yamle.models.resnet.BasicBlock method) (yamle.models.resnet.Blocks method) (yamle.models.resnet.Bottleneck method) (yamle.models.resnet.ResNetModel method) (yamle.models.rnn.InputRepeater method) (yamle.models.rnn.RNNAutoEncoderModel method) (yamle.models.rnn.RNNModel method) (yamle.models.transformer.Attention method) (yamle.models.transformer.PositionalEncoding method) (yamle.models.transformer.TransformerModel method) (yamle.models.unet.DownBlock method) (yamle.models.unet.UNetModel method) (yamle.models.unet.UpBlock method) (yamle.models.vgg.Blocks method) (yamle.models.vgg.EmptyBlock method) (yamle.models.vgg.VGGBlock method) (yamle.models.vgg.VGGModel method) (yamle.models.visual_transformer.SpatialPositionalEmbedding method) (yamle.models.visual_transformer.VisualTransformerModel method) (yamle.quantization.models.operations.QuantizableAdd method) (yamle.quantization.models.specific.mcdropout.QuantisedDropout1d method) (yamle.quantization.models.specific.mcdropout.QuantisedDropout2d method) (yamle.quantization.models.specific.mcdropout.QuantisedDropout3d method)

forward_shape_hook() (in module yamle.utils.tracing_utils) freeze_weights() (in module yamle.utils.optimization_utils) frequency (yamle.utils.trainer_utils.GradientNormMonitorCallback parameter) FromOneHot (class in yamle.data.transforms) frost() (in module yamle.third_party.imagenet_c.corruptions) fullrank (yamle.methods.swag.SWAGMethod parameter)

G

gamma (yamle.losses.segmentation.FocalLoss parameter) (yamle.methods.early_exit.EarlyExitMethod parameter) (yamle.methods.pe.PEMethod parameter) (yamle.methods.pe.replace_layers_with_grouped_convs parameter) (yamle.regularizers.gradient.GradientNoiseRegularizer parameter) (yamle.utils.optimization_utils.PowerGrowthScalarScheduler parameter) gating_depth (yamle.methods.moe.MixtureOfExpertsMethod parameter) gating_expansion_factor (yamle.methods.moe.MixtureOfExpertsMethod parameter) gaussian_blur() (in module yamle.third_party.imagenet_c.corruptions) gaussian_noise() (in module yamle.third_party.imagenet_c.corruptions) GaussianNegativeLogLikelihoodLoss (class in yamle.losses.regression) generate() (yamle.models.transformer.TransformerModel method) get_all_prunable_modules() (in module yamle.utils.pruning_utils) get_all_prunable_weights() (in module yamle.utils.pruning_utils) get_cls_token_indices() (yamle.models.visual_transformer.SpatialPositionalEmbedding method) get_experiment_name() (in module yamle.utils.file_utils) get_input_shape_from_model() (in module yamle.utils.tracing_utils) get_named_parameters() (yamle.methods.method.BaseMethod method) get_names() (yamle.regularizers.regularizer.BaseRegularizer method) get_optimizer() (in module yamle.utils.optimization_utils) get_output_shape_from_model() (in module yamle.utils.tracing_utils) get_parameters() (yamle.methods.ensemble.SnapsotEnsembleMethod method) (yamle.methods.gp.GPMethod method) (yamle.methods.method.BaseMethod method) (yamle.methods.sgld.SGLDMethod method) (yamle.regularizers.regularizer.BaseRegularizer method) get_qconfig() (yamle.quantization.qat.QATQuantizer method) (yamle.quantization.quantizer.BaseQuantizer method) (yamle.quantization.quantizer.DummyQuantizer method) (yamle.quantization.static.StaticQuantizer method)

get_sample_input_and_target() (in module yamle.utils.tracing_utils) get_scheduler() (in module yamle.utils.optimization_utils) get_transform() (yamle.data.custom.ECG5000DataModule method) (yamle.data.datamodule.BaseDataModule method) (yamle.data.datamodule.RealWorldDataModule method) (yamle.data.datamodule.VisionDataModule method) (yamle.data.depth.DepthEstimationDataModule method) (yamle.data.segmentation.TorchvisionSegmentationDataModule method) get_transform_composition() (yamle.data.datamodule.BaseDataModule method) get_value() (yamle.utils.optimization_utils.CosineScalarScheduler method) (yamle.utils.optimization_utils.LinearScalarScheduler method) (yamle.utils.optimization_utils.PowerGrowthScalarScheduler method) (yamle.utils.optimization_utils.ScalarScheduler method) (yamle.utils.optimization_utils.SineScalarScheduler method) glass_blur() (in module yamle.third_party.imagenet_c.corruptions) GPMethod (class in yamle.methods.gp) GPModel (class in yamle.models.gp) gradient_clip_norm_value (yamle.trainers.trainer.BaseTrainer parameter) gradient_clip_value (yamle.trainers.trainer.BaseTrainer parameter) GradientBoostingEnsembleMethod (class in yamle.methods.ensemble) GradientNoiseRegularizer (class in yamle.regularizers.gradient) GradientNormMonitorCallback (class in yamle.utils.trainer_utils) GradientValueClippingCallback (class in yamle.utils.trainer_utils) groups (yamle.models.operations.Conv2dNormActivation parameter) (yamle.models.resnet.Bottleneck attribute) growth_rate (yamle.models.densenet.DenseBlock parameter) (yamle.models.densenet.DenseLayer parameter) (yamle.models.densenet.DenseNetModel parameter)

H

h (yamle.models.operations.LSTM.forward parameter) head_depth (yamle.methods.moe.MultiHeadEnsembleMethod parameter) head_expansion_factor (yamle.methods.moe.MultiHeadEnsembleMethod parameter) heads (yamle.models.transformer.Attention parameter) (yamle.models.transformer.TransformerEncoderLayer parameter) height (yamle.data.transforms.JointCenterCrop parameter) (yamle.data.transforms.JointResize parameter) hessian_computed (yamle.methods.laplace.LaplaceMethod property)

hidden_dim (yamle.models.fc.FCModel parameter) (yamle.models.operations.Conv2dExtractor parameter) (yamle.models.operations.LinearExtractor parameter) (yamle.models.rnn.RNNAutoEncoderModel parameter) (yamle.models.rnn.RNNModel parameter) (yamle.models.transformer.FeedForward parameter) (yamle.models.transformer.TransformerModel parameter) (yamle.models.visual_transformer.VisualTransformerModel parameter) hidden_size (yamle.models.operations.LSTM parameter)

I

ignore_arguments (yamle.utils.file_utils.argparse_to_config_file parameter) ignore_indices (yamle.data.datamodule.BaseDataModule attribute) (yamle.data.segmentation.TorchvisionSegmentationDataModuleCityscapes attribute) (yamle.losses.segmentation.FocalLoss parameter) (yamle.losses.segmentation.SoftIntersectionOverUnionLoss parameter) image_size (yamle.data.regression.WikiFaceDataset parameter) ImageRotationDataset (class in yamle.data.dataset_wrappers) img_dtype (yamle.data.transforms.JointToTensor parameter) img_size (yamle.third_party.imagenet_c.VisionCorruption parameter) import_config_function_from_file() (in module yamle.utils.file_utils) import_lines (yamle.utils.file_utils.argparse_to_config_file parameter) impulse_noise() (in module yamle.third_party.imagenet_c.corruptions) in_channels (yamle.models.operations.CompletelySeparableConv2d parameter) (yamle.models.operations.Conv2dNormActivation parameter) (yamle.models.operations.DepthwiseSeparableConv2d parameter) (yamle.models.operations.DoubleConv2d parameter) (yamle.models.unet.DownBlock parameter) (yamle.models.unet.UpBlock parameter) in_features (yamle.models.operations.LinearNormActivation parameter) in_out_channels (yamle.models.operations.SqueezeAndExcitation parameter) increment_current_member() (yamle.methods.ensemble.EnsembleMethod method) indices (yamle.data.transforms.ClassificationDatasetSubset parameter) inducing_points (yamle.models.gp.GPModel parameter) init_features (yamle.models.unet.UNetModel parameter) initial_p (yamle.methods.mimo.MixMoMethod parameter) initial_planes (yamle.models.densenet.DenseNetModel parameter) initialise_encoder_members_same (yamle.methods.mimo.MIMOMethod parameter) initialise_members_same (yamle.models.operations.ParallelModel parameter) initialise_members_same() (yamle.models.operations.ParallelModel method) InitialModelSavingCallback (class in yamle.utils.trainer_utils) InnerProductFeatureRegularizer (class in yamle.regularizers.feature) inplace (yamle.models.operations.Add parameter) (yamle.models.operations.Multiply parameter) (yamle.models.operations.ResidualLayer parameter) (yamle.quantization.models.specific.mcdropout.QuantisedDropout1d parameter) (yamle.quantization.models.specific.mcdropout.QuantisedDropout2d parameter) (yamle.quantization.models.specific.mcdropout.QuantisedDropout3d parameter) inplanes (yamle.models.densenet.DenseBlock parameter) (yamle.models.densenet.DenseLayer parameter) (yamle.models.densenet.Transition parameter) (yamle.models.resnet.BasicBlock parameter) (yamle.models.resnet.Blocks parameter) (yamle.models.resnet.Bottleneck parameter) (yamle.models.vgg.Blocks parameter) (yamle.models.vgg.VGGBlock parameter) input (yamle.utils.tracing_utils.forward_shape_hook parameter) input_channels (yamle.models.operations.Conv2dExtractor parameter) input_kwargs (yamle.models.convnet.ConvNetModel.forward parameter) (yamle.models.densenet.DenseNetModel.forward parameter) (yamle.models.fc.FCModel.forward parameter) (yamle.models.gp.GPModel.forward parameter) (yamle.models.mixer.MixerModel.forward parameter) (yamle.models.resnet.ResNetModel.forward parameter) (yamle.models.rnn.RNNAutoEncoderModel.forward parameter) (yamle.models.rnn.RNNModel.forward parameter) (yamle.models.transformer.TransformerModel.forward parameter) (yamle.models.unet.UNetModel.forward parameter) (yamle.models.vgg.VGGModel.forward parameter) (yamle.models.visual_transformer.VisualTransformerModel.forward parameter) input_repetition_probability (yamle.methods.mimo.MIMOMethod parameter) input_size (yamle.models.operations.LSTM parameter) InputImagePaddingDataset (class in yamle.data.dataset_wrappers) InputRepeater (class in yamle.models.rnn) inputs (yamle.data.augmentations.CutMix.augment parameter) (yamle.data.augmentations.CutOut.augment parameter) (yamle.data.augmentations.RandomErasing.augment parameter) (yamle.models.densenet.DenseBlock.forward parameter) (yamle.models.densenet.DenseLayer.bn_function parameter) (yamle.models.densenet.DenseLayer.forward parameter) (yamle.models.densenet.EmptyBlock.forward parameter) (yamle.models.densenet.Transition.forward parameter) inputs_dim (yamle.data.classification.AdultIncomeUCIClassificationDataModule attribute) (yamle.data.classification.BloodMNISTClassificationDataModule attribute) (yamle.data.classification.BreastCancerUCIClassificationDataModule attribute) (yamle.data.classification.BreastMNISTClassificationDataModule attribute) (yamle.data.classification.CarEvaluationUCIClassificationDataModule attribute) (yamle.data.classification.CreditUCIClassificationDataModule attribute) (yamle.data.classification.DermaMNISTClassificationDataModule attribute) (yamle.data.classification.DermatologyUCIClassificationDataModule attribute) (yamle.data.classification.PneumoniaMNISTClassificationDataModule attribute) (yamle.data.classification.TinyImageNetClassificationDataModule attribute) (yamle.data.classification.TorchvisionClassificationDataModuleCIFAR10 attribute) (yamle.data.classification.TorchvisionClassificationDataModuleCIFAR100 attribute) (yamle.data.classification.TorchvisionClassificationDataModuleCIFAR3 attribute) (yamle.data.classification.TorchvisionClassificationDataModuleCIFAR5 attribute) (yamle.data.classification.TorchvisionClassificationDataModuleFashionMNIST attribute) (yamle.data.classification.TorchvisionClassificationDataModuleMNIST attribute) (yamle.data.classification.TorchvisionClassificationDataModuleSVHN attribute) (yamle.data.classification.ToyTwoMoonsClassificationDataModule attribute) (yamle.data.custom.ECG5000DataModule attribute) (yamle.data.datamodule.BaseDataModule attribute) (yamle.data.depth.DepthEstimationDataModule attribute) (yamle.data.depth.NYUv2DataModule attribute) (yamle.data.regression.AbaloneUCIRegressionDataModule attribute) (yamle.data.regression.BostonUCIRegressionDataModule attribute) (yamle.data.regression.ConcreteUCIRegressionDataModule attribute) (yamle.data.regression.EnergyUCIRegressionDataModule attribute) (yamle.data.regression.RetinaMNISTDataModule attribute) (yamle.data.regression.TelemonitoringUCIRegressionDataModule attribute) (yamle.data.regression.TemperatureTimeSeriesDataModule attribute) (yamle.data.regression.TinyImageNetRotationRegressionDataModule attribute) (yamle.data.regression.TorchvisionRotationRegressionDataModuleCIFAR10 attribute) (yamle.data.regression.TorchvisionRotationRegressionDataModuleCIFAR100 attribute) (yamle.data.regression.TorchvisionRotationRegressionDataModuleFashionMNIST attribute) (yamle.data.regression.TorchvisionRotationRegressionDataModuleMNIST attribute) (yamle.data.regression.TorchvisionRotationRegressionDataModuleSVHN attribute) (yamle.data.regression.ToyRegressionDataModule attribute) (yamle.data.regression.WikiFaceRegressionDataModule attribute) (yamle.data.regression.WineQualityUCIRegressionDataModule attribute) (yamle.data.regression.YachtUCIRegressionDataModule attribute) (yamle.data.segmentation.TorchvisionSegmentationDataModule attribute) (yamle.data.segmentation.TorchvisionSegmentationDataModuleCityscapes attribute) (yamle.data.text.Shakespeare attribute) (yamle.data.text.TorchtextClassificationDataModule attribute) (yamle.data.text.TorchtextClassificationModelIMDB attribute) (yamle.data.text.TorchtextClassificationModelWikiText103 attribute) (yamle.data.text.TorchtextClassificationModelWikiText2 attribute) (yamle.methods.method.BaseMethod parameter) (yamle.models.convnet.ConvNetModel parameter) (yamle.models.model.BaseModel parameter) (yamle.models.operations.LinearExtractor parameter) (yamle.models.operations.ParallelModel parameter) (yamle.models.transformer.PositionalEncoding parameter) (yamle.models.visual_transformer.SpatialPositionalEmbedding parameter)

inputs_dtype (yamle.data.classification.AdultIncomeUCIClassificationDataModule attribute) (yamle.data.classification.BreastCancerUCIClassificationDataModule attribute) (yamle.data.classification.CarEvaluationUCIClassificationDataModule attribute) (yamle.data.classification.CreditUCIClassificationDataModule attribute) (yamle.data.classification.DermatologyUCIClassificationDataModule attribute) (yamle.data.classification.ECG5000ClassificationDataModule attribute) (yamle.data.classification.ToyTwoMoonsClassificationDataModule attribute) (yamle.data.datamodule.BaseDataModule attribute) (yamle.data.datamodule.RealWorldClassificationDataModule attribute) (yamle.data.datamodule.RealWorldDataModule attribute) (yamle.data.datamodule.RealWorldRegressionDataModule attribute) (yamle.data.datamodule.VisionDataModule attribute) (yamle.data.depth.DepthEstimationDataModule attribute) (yamle.data.reconstruction.ECG5000ReconstructionDataModule attribute) (yamle.data.regression.UCIRegressionDataModule attribute) (yamle.data.segmentation.TorchvisionSegmentationDataModule attribute) (yamle.data.text.TorchtextClassificationDataModule attribute) (yamle.methods.method.BaseMethod parameter) interrupted (yamle.trainers.trainer.BaseTrainer property) inverse() (yamle.data.transforms.Normalize method) is_layer_prunable() (in module yamle.utils.pruning_utils) is_parameter_prunable() (in module yamle.utils.pruning_utils)

J

joint_transform (yamle.data.dataset_wrappers.SurrogateDataset parameter) JointCenterCrop (class in yamle.data.transforms) JointCompose (class in yamle.data.transforms) JointNormalize (class in yamle.data.transforms)

JointResize (class in yamle.data.transforms) JointTargetSqueeze (class in yamle.data.transforms) JointToTensor (class in yamle.data.transforms) jpeg_compression() (in module yamle.third_party.imagenet_c.corruptions)

K

k (yamle.methods.moe.MixtureOfExpertsMethod parameter) kernel_size (yamle.models.operations.CompletelySeparableConv2d parameter) (yamle.models.operations.Conv2dNormActivation parameter) (yamle.models.operations.DepthwiseSeparableConv2d parameter) (yamle.models.operations.DoubleConv2d parameter)

L

l (yamle.regularizers.model.ShrinkAndPerturbRegularizer parameter) L1FeatureRegularizer (class in yamle.regularizers.feature) L1L2MonitorCallback (class in yamle.utils.trainer_utils) L1L2Regularizer (class in yamle.regularizers.weight) L1Regularizer (class in yamle.regularizers.weight) L2FeatureRegularizer (class in yamle.regularizers.feature) L2Regularizer (class in yamle.regularizers.weight) label_smoothing (yamle.losses.classification.CrossEntropyLoss parameter) (yamle.losses.classification.TextCrossEntropyLoss parameter) lam (yamle.data.augmentations.rand_bbox parameter) Lambda (class in yamle.models.operations) LaplaceMethod (class in yamle.methods.laplace) layer (yamle.models.operations.ResidualLayer parameter) (yamle.utils.pruning_utils.is_layer_prunable parameter) layers (yamle.models.densenet.DenseNetModel parameter) (yamle.models.resnet.ResNetModel parameter) (yamle.models.vgg.VGGModel parameter) learning_rate (yamle.methods.method.BaseMethod parameter) (yamle.quantization.qat.QATQuantizer parameter) linear_hidden_dims (yamle.models.convnet.ConvNetModel parameter) LinearExtractor (class in yamle.models.operations) LinearNormActivation (class in yamle.models.operations) LinearScalarScheduler (class in yamle.utils.optimization_utils) load_args() (in module yamle.utils.file_utils) load_args_dictionary() (in module yamle.utils.file_utils) load_method() (in module yamle.utils.file_utils) load_model() (in module yamle.utils.file_utils)

load_pickle() (in module yamle.utils.file_utils) load_results() (in module yamle.utils.file_utils) load_state_dict() (yamle.methods.delta_uq.DeltaUQMethod method) (yamle.methods.dun.DUNMethod method) (yamle.methods.gp.GPMethod method) (yamle.methods.laplace.LaplaceMethod method) (yamle.methods.method.BaseMethod method) (yamle.methods.mimo.MIMMOMethod method) (yamle.methods.mimo.MIMOMethod method) (yamle.methods.mimo.UnMixMoMethod method) (yamle.methods.swag.SWAGMethod method) (yamle.utils.optimization_utils.CosineScalarScheduler method) (yamle.utils.optimization_utils.LinearScalarScheduler method) (yamle.utils.optimization_utils.PowerGrowthScalarScheduler method) (yamle.utils.optimization_utils.ScalarScheduler method) (yamle.utils.optimization_utils.SineScalarScheduler method) load_tuning_results() (in module yamle.utils.file_utils) log_file() (in module yamle.utils.file_utils) log_variance (yamle.methods.rbnn.RBNNMethod parameter) (yamle.methods.rbnn.replace_with_rbnn parameter) (yamle.methods.svi.SVIReparameterizationMethod parameter) LoggingCallback (class in yamle.utils.trainer_utils) loss (yamle.methods.method.BaseMethod parameter) (yamle.methods.method.BaseMethod.backward parameter) loss_factory() (in module yamle.losses) loss_type (yamle.losses.loss_factory parameter) LSTM (class in yamle.models.operations)

M

M (yamle.methods.pe.replace_layers_with_grouped_convs parameter) m (yamle.methods.sngp.SNGPMethod parameter) (yamle.utils.pruning_utils.disable_pruning parameter) (yamle.utils.pruning_utils.enable_pruning parameter) (yamle.utils.pruning_utils.get_all_prunable_modules parameter) (yamle.utils.pruning_utils.get_all_prunable_weights parameter) m_end_epoch (yamle.methods.mimo.UnMixMoMethod parameter) m_end_value (yamle.methods.mimo.UnMixMoMethod parameter) m_start_epoch (yamle.methods.mimo.UnMixMoMethod parameter) m_start_value (yamle.methods.mimo.UnMixMoMethod parameter) masks (yamle.utils.optimization_utils.freeze_weights parameter) MatrixMultiplication (class in yamle.models.operations) max_angle (yamle.data.dataset_wrappers.ImageRotationDataset parameter) (yamle.data.regression.TorchvisionRotationRegressionDataModule parameter) max_len (yamle.models.transformer.PositionalEncoding parameter) maximum (yamle.third_party.imagenet_c.extra.RandomImageNoise parameter) (yamle.third_party.imagenet_c.extra.RandomTabularNoise parameter) MCDropBlockMethod (class in yamle.methods.mcdropout) MCDropConnectMethod (class in yamle.methods.mcdropout) MCDropoutMethod (class in yamle.methods.mcdropout) MCSamplingMethod (class in yamle.methods.uncertain_method) MCStandOutMethod (class in yamle.methods.mcdropout) MCStochasticDepthMethod (class in yamle.methods.mcdropout) md5sum (yamle.data.regression.WikiFaceDataset attribute) mean (yamle.data.classification.BloodMNISTClassificationDataModule attribute) (yamle.data.classification.BreastMNISTClassificationDataModule attribute) (yamle.data.classification.DermaMNISTClassificationDataModule attribute) (yamle.data.classification.PneumoniaMNISTClassificationDataModule attribute) (yamle.data.classification.TinyImageNetClassificationDataModule attribute) (yamle.data.classification.TorchvisionClassificationDataModuleCIFAR10 attribute) (yamle.data.classification.TorchvisionClassificationDataModuleCIFAR100 attribute) (yamle.data.classification.TorchvisionClassificationDataModuleCIFAR3 attribute) (yamle.data.classification.TorchvisionClassificationDataModuleCIFAR5 attribute) (yamle.data.classification.TorchvisionClassificationDataModuleFashionMNIST attribute) (yamle.data.classification.TorchvisionClassificationDataModuleMNIST attribute) (yamle.data.classification.TorchvisionClassificationDataModuleSVHN attribute) (yamle.data.datamodule.VisionDataModule attribute) (yamle.data.depth.DepthEstimationDataModule attribute) (yamle.data.depth.NYUv2DataModule attribute) (yamle.data.regression.RetinaMNISTDataModule attribute) (yamle.data.regression.TinyImageNetRotationRegressionDataModule attribute) (yamle.data.regression.TorchvisionRotationRegressionDataModuleCIFAR10 attribute) (yamle.data.regression.TorchvisionRotationRegressionDataModuleCIFAR100 attribute) (yamle.data.regression.TorchvisionRotationRegressionDataModuleFashionMNIST attribute) (yamle.data.regression.TorchvisionRotationRegressionDataModuleMNIST attribute) (yamle.data.regression.TorchvisionRotationRegressionDataModuleSVHN attribute) (yamle.data.regression.WikiFaceRegressionDataModule attribute) (yamle.data.segmentation.TorchvisionSegmentationDataModule attribute) (yamle.data.segmentation.TorchvisionSegmentationDataModuleCityscapes attribute) (yamle.data.text.TorchtextClassificationDataModule attribute) (yamle.data.transforms.Denormalize parameter) (yamle.data.transforms.JointNormalize parameter) (yamle.data.transforms.Normalize parameter) (yamle.third_party.imagenet_c.extra.RandomImageNoise parameter) (yamle.third_party.imagenet_c.extra.RandomTabularNoise parameter) mean_field_factor (yamle.methods.sngp.SNGPMethod parameter) MeanSquaredError (class in yamle.losses.regression) MedMNISTClassificationDataModule (class in yamle.data.classification) MedMNISTDatasetWrapper (class in yamle.third_party.medmnist) MedMNISTRegressionDataModule (class in yamle.data.regression) MemberMethod (class in yamle.methods.uncertain_method) method (yamle.methods.rbnn.RBNNMethod parameter) (yamle.methods.rbnn.replace_with_rbnn parameter) (yamle.methods.svi.SVIReparameterizationMethod parameter) (yamle.models.convnet.ConvNetModel.add_method_specific_layers parameter) (yamle.models.densenet.DenseNetModel.add_method_specific_layers parameter) (yamle.models.fc.FCModel.add_method_specific_layers parameter) (yamle.models.gp.GPModel.add_method_specific_layers parameter) (yamle.models.mixer.MixerModel.add_method_specific_layers parameter) (yamle.models.model.BaseModel.add_method_specific_layers parameter) (yamle.models.resnet.ResNetModel.add_method_specific_layers parameter) (yamle.models.rnn.RNNModel.add_method_specific_layers parameter) (yamle.models.transformer.TransformerModel.add_method_specific_layers parameter) (yamle.models.unet.UNetModel.add_method_specific_layers parameter) (yamle.models.vgg.VGGModel.add_method_specific_layers parameter) (yamle.models.visual_transformer.VisualTransformerModel.add_method_specific_layers parameter) (yamle.trainers.trainer.BaseTrainer parameter) (yamle.utils.file_utils.save_experiment parameter) method_factory() (in module yamle.methods) method_file() (in module yamle.utils.file_utils) method_type (yamle.methods.method_factory parameter) metric (yamle.utils.plotting_utils.average_augmentations parameter) metrics (yamle.utils.file_utils.store_metrics parameter) (yamle.utils.plotting_utils.pareto_optimal parameter) metrics_kwargs (yamle.methods.method.BaseMethod parameter) MIMMOMethod (class in yamle.methods.mimo) MIMOMethod (class in yamle.methods.mimo) min_angle (yamle.data.dataset_wrappers.ImageRotationDataset parameter) (yamle.data.regression.TorchvisionRotationRegressionDataModule parameter) minimum (yamle.third_party.imagenet_c.extra.RandomImageNoise parameter) (yamle.third_party.imagenet_c.extra.RandomTabularNoise parameter) MixerLayer (class in yamle.models.mixer) MixerModel (class in yamle.models.mixer) MixMoMethod (class in yamle.methods.mimo) MixtureOfExpertsMethod (class in yamle.methods.moe) MixUp (class in yamle.data.augmentations) mixup_alpha (yamle.methods.augmentation_classification.MixUpImageClassificationMethod parameter) MixUpImageClassificationMethod (class in yamle.methods.augmentation_classification) MixVitMethod (class in yamle.methods.mimo) mlp_dim (yamle.models.transformer.TransformerEncoderLayer parameter) mode (yamle.methods.laplace.LaplaceMethod parameter) (yamle.methods.mcdropout.MCDropBlockMethod parameter) (yamle.methods.mcdropout.MCDropConnectMethod parameter) (yamle.methods.mcdropout.MCDropoutMethod parameter) (yamle.methods.mcdropout.MCStandOutMethod parameter) (yamle.methods.svi.SVIReparameterizationMethod parameter) (yamle.trainers.trainer.BaseTrainer parameter) model (yamle.data.augmentations.CutMix.augment parameter) (yamle.data.augmentations.CutOut.augment parameter) (yamle.data.augmentations.RandomErasing.augment parameter) (yamle.methods.be.replace_with_be parameter) (yamle.methods.method.BaseMethod parameter) (yamle.methods.pe.replace_layers_with_grouped_convs parameter) (yamle.methods.rbnn.replace_with_rbnn parameter) (yamle.methods.sngp.enable_spectral_normalization parameter) (yamle.utils.export_utils.export_onnx parameter) (yamle.utils.file_utils.save_experiment parameter) (yamle.utils.optimization_utils.recover_frozen_weights parameter) model_best_on_validation_file() (in module yamle.utils.file_utils) model_factory() (in module yamle.models) model_file() (in module yamle.utils.file_utils)

model_initial_file() (in module yamle.utils.file_utils) model_kwargs (yamle.methods.method.BaseMethod parameter) model_onnx_file() (in module yamle.utils.file_utils) model_quantized_file() (in module yamle.utils.file_utils) model_train_epoch_file() (in module yamle.utils.file_utils) model_type (yamle.models.model_factory parameter) models (yamle.models.operations.ParallelModel parameter) module yamle yamle.data yamle.data.augmentations yamle.data.classification yamle.data.custom yamle.data.datamodule yamle.data.dataset_wrappers yamle.data.depth yamle.data.reconstruction yamle.data.regression yamle.data.segmentation yamle.data.text yamle.data.transforms yamle.defaults yamle.losses yamle.losses.classification yamle.losses.contrastive yamle.losses.evidential_regression yamle.losses.loss yamle.losses.regression yamle.losses.segmentation yamle.methods yamle.methods.augmentation_classification yamle.methods.be yamle.methods.contrastive yamle.methods.delta_uq yamle.methods.dun yamle.methods.early_exit yamle.methods.ensemble yamle.methods.evidential_regression yamle.methods.gp yamle.methods.laplace yamle.methods.mcdropout yamle.methods.method yamle.methods.mimo yamle.methods.moe yamle.methods.pe yamle.methods.rbnn yamle.methods.sgld yamle.methods.sngp yamle.methods.svi yamle.methods.swag yamle.methods.temperature_scaling yamle.methods.uncertain_method yamle.models yamle.models.convnet yamle.models.densenet yamle.models.fc yamle.models.gp yamle.models.mixer yamle.models.model yamle.models.operations yamle.models.resnet yamle.models.rnn yamle.models.transformer yamle.models.unet yamle.models.vgg yamle.models.visual_transformer yamle.pruning yamle.pruning.pruner yamle.pruning.unstructured yamle.pruning.unstructured.magnitude yamle.quantization yamle.quantization.models yamle.quantization.models.operations yamle.quantization.models.specific yamle.quantization.models.specific.mcdropout yamle.quantization.qat yamle.quantization.quantizer yamle.quantization.static yamle.regularizers yamle.regularizers.feature yamle.regularizers.gradient yamle.regularizers.model yamle.regularizers.regularizer yamle.regularizers.weight yamle.third_party yamle.third_party.imagenet_c yamle.third_party.imagenet_c.corruptions yamle.third_party.imagenet_c.extra yamle.third_party.imagenet_c.tabular_corruptions yamle.third_party.medmnist yamle.third_party.nyuv2 yamle.third_party.tinyimagenet yamle.trainers yamle.trainers.calibration yamle.trainers.ensemble yamle.trainers.trainer yamle.utils yamle.utils.cli_utils yamle.utils.export_utils yamle.utils.file_utils yamle.utils.operation_utils yamle.utils.optimization_utils yamle.utils.plotting_utils yamle.utils.pruning_utils yamle.utils.regularizer_utils yamle.utils.running_utils yamle.utils.specific yamle.utils.specific.mimo_experiments yamle.utils.specific.mimo_experiments.plotting_utils yamle.utils.tracing_utils yamle.utils.trainer_utils yamle.utils.tuning_utils module (yamle.models.transformer.PreNorm parameter) (yamle.utils.tracing_utils.forward_shape_hook parameter) momentum (yamle.methods.method.BaseMethod parameter) motion_blur() (in module yamle.third_party.imagenet_c.corruptions) MultiHeadEnsembleMethod (class in yamle.methods.moe) multiplicative_bernoulli_noise() (in module yamle.third_party.imagenet_c.tabular_corruptions) multiplicative_gaussian_noise() (in module yamle.third_party.imagenet_c.tabular_corruptions) multiplicative_uniform_noise() (in module yamle.third_party.imagenet_c.tabular_corruptions) Multiply (class in yamle.models.operations)

N

n_layers (yamle.models.densenet.DenseBlock parameter) name_all_modules() (in module yamle.utils.tracing_utils) NIG_NLL (class in yamle.losses.evidential_regression) NIG_Reg (class in yamle.losses.evidential_regression) no_augmentation_testing (yamle.trainers.trainer.BaseTrainer parameter) no_every_train_epoch_saving (yamle.trainers.trainer.BaseTrainer parameter) no_initial_saving (yamle.trainers.trainer.BaseTrainer parameter) no_input_replacement (yamle.methods.mcdropout.MCDropBlockMethod parameter) (yamle.methods.mcdropout.MCDropConnectMethod parameter) (yamle.methods.mcdropout.MCDropoutMethod parameter) (yamle.methods.mcdropout.MCStandOutMethod parameter) no_saving (yamle.trainers.trainer.BaseTrainer parameter) no_validation_saving (yamle.trainers.trainer.BaseTrainer parameter) noise (yamle.data.classification.ToyTwoMoonsClassificationDataModule parameter) (yamle.third_party.imagenet_c.extra.RandomImageNoise parameter) (yamle.third_party.imagenet_c.extra.RandomTabularNoise parameter) NoiseContrastiveEstimatorLoss (class in yamle.losses.contrastive) noisy_gate (yamle.methods.moe.MixtureOfExpertsMethod parameter) NoOptimizationMonitorCallback (class in yamle.utils.trainer_utils) norm (yamle.models.operations.Conv2dExtractor parameter) (yamle.models.operations.LinearExtractor parameter) (yamle.models.operations.Normalization parameter) (yamle.utils.trainer_utils.GradientNormMonitorCallback parameter) norm_kwargs (yamle.models.operations.Normalization parameter) Normalization (class in yamle.models.operations) normalization (yamle.models.convnet.ConvNetModel parameter) (yamle.models.densenet.DenseBlock parameter) (yamle.models.densenet.DenseLayer parameter) (yamle.models.densenet.DenseNetModel parameter) (yamle.models.densenet.Transition parameter) (yamle.models.fc.FCModel parameter) (yamle.models.operations.CompletelySeparableConv2d parameter) (yamle.models.operations.Conv2dExtractor parameter) (yamle.models.operations.Conv2dNormActivation parameter) (yamle.models.operations.DepthwiseSeparableConv2d parameter) (yamle.models.operations.DoubleConv2d parameter) (yamle.models.operations.LinearExtractor parameter) (yamle.models.operations.LinearNormActivation parameter) (yamle.models.resnet.BasicBlock parameter) (yamle.models.resnet.Blocks parameter) (yamle.models.resnet.Bottleneck parameter) (yamle.models.resnet.ResNetModel parameter) (yamle.models.unet.DownBlock parameter) (yamle.models.unet.UNetModel parameter) (yamle.models.unet.UpBlock parameter) (yamle.models.vgg.Blocks parameter) (yamle.models.vgg.VGGBlock parameter) (yamle.models.vgg.VGGModel parameter)

normalization_kwargs (yamle.models.densenet.DenseBlock parameter) (yamle.models.densenet.DenseLayer parameter) (yamle.models.densenet.Transition parameter) (yamle.models.resnet.BasicBlock parameter) (yamle.models.resnet.Blocks parameter) (yamle.models.resnet.Bottleneck parameter) (yamle.models.vgg.Blocks parameter) (yamle.models.vgg.VGGBlock parameter) Normalize (class in yamle.data.transforms) normalize_tin_validation_folder_structure() (in module yamle.third_party.tinyimagenet) num_batch_repetitions (yamle.methods.mimo.MIMOMethod parameter) num_cls_tokens (yamle.models.visual_transformer.SpatialPositionalEmbedding parameter) (yamle.models.visual_transformer.VisualTransformerModel parameter) num_decoder_layers (yamle.models.transformer.TransformerModel parameter) num_heads (yamle.models.transformer.TransformerModel parameter) (yamle.models.visual_transformer.VisualTransformerModel parameter) num_inducing_points (yamle.methods.gp.GPMethod parameter) num_latent (yamle.methods.gp.GPMethod parameter) (yamle.models.gp.GPModel parameter) num_layers (yamle.models.mixer.MixerModel parameter) num_members (yamle.methods.be.replace_with_be parameter) (yamle.methods.ensemble.EnsembleMethod parameter) (yamle.methods.ensemble.GradientBoostingEnsembleMethod parameter) (yamle.methods.ensemble.SnapsotEnsembleMethod parameter) (yamle.methods.rbnn.RBNNMethod parameter) (yamle.methods.rbnn.replace_with_rbnn parameter) (yamle.methods.swag.SWAGMethod parameter) (yamle.methods.uncertain_method.MemberMethod parameter) num_outputs (yamle.models.gp.GPModel parameter) num_samples (yamle.data.classification.ToyTwoMoonsClassificationDataModule parameter) (yamle.data.regression.ToyRegressionDataModule parameter) num_workers (yamle.data.datamodule.BaseDataModule parameter) NYUv2 (class in yamle.third_party.nyuv2) NYUv2DataModule (class in yamle.data.depth)

O

on_after_backward() (yamle.methods.method.BaseMethod method) (yamle.regularizers.gradient.GradientNoiseRegularizer method) (yamle.regularizers.regularizer.BaseRegularizer method) (yamle.utils.trainer_utils.GradientNormMonitorCallback method) (yamle.utils.trainer_utils.GradientValueClippingCallback method) on_after_method_load() (yamle.methods.method.BaseMethod method) on_after_model_load() (yamle.methods.laplace.LaplaceMethod method) (yamle.methods.method.BaseMethod method) on_after_train_epoch() (yamle.regularizers.model.ShrinkAndPerturbRegularizer method) (yamle.regularizers.regularizer.BaseRegularizer method) on_after_training_step() (yamle.regularizers.regularizer.BaseRegularizer method) (yamle.regularizers.weight.WeightDecayRegularizer method) on_before_backward() (yamle.methods.method.BaseMethod method) on_before_method_load() (yamle.methods.method.BaseMethod method) on_before_model_load() (yamle.methods.method.BaseMethod method) on_fit_end() (yamle.methods.laplace.LaplaceMethod method) on_fit_start() (yamle.methods.method.BaseMethod method) (yamle.utils.trainer_utils.InitialModelSavingCallback method) (yamle.utils.trainer_utils.NoOptimizationMonitorCallback method) (yamle.utils.trainer_utils.RegularizedWeightsMonitorCallback method) (yamle.utils.trainer_utils.SplitParametersMonitorCallback method) on_test_batch_end() (yamle.methods.method.BaseMethod method) (yamle.utils.trainer_utils.LoggingCallback method) on_test_epoch_end() (yamle.methods.method.BaseMethod method) (yamle.utils.trainer_utils.LoggingCallback method) on_test_epoch_start() (yamle.methods.method.BaseMethod method) (yamle.utils.trainer_utils.LoggingCallback method) on_test_start() (yamle.methods.method.BaseMethod method) on_train_batch_end() (yamle.methods.method.BaseMethod method) (yamle.utils.trainer_utils.LoggingCallback method) on_train_epoch_end() (yamle.methods.dun.DUNMethod method) (yamle.methods.ensemble.SnapsotEnsembleMethod method) (yamle.methods.method.BaseMethod method) (yamle.methods.mimo.MIMMOMethod method) (yamle.methods.mimo.MIMOMethod method) (yamle.methods.mimo.UnMixMoMethod method) (yamle.methods.sgld.SGLDMethod method) (yamle.methods.sngp.SNGPMethod method) (yamle.methods.swag.SWAGMethod method) (yamle.utils.trainer_utils.L1L2MonitorCallback method) (yamle.utils.trainer_utils.LoggingCallback method) (yamle.utils.trainer_utils.TrainingModelSavingCallback method) (yamle.utils.trainer_utils.WeightHistogramMonitorCallback method) on_train_epoch_start() (yamle.methods.method.BaseMethod method) (yamle.methods.mimo.MixMoMethod method) (yamle.methods.mimo.UnMixMoMethod method) (yamle.methods.sngp.SNGPMethod method) (yamle.methods.swag.SWAGMethod method) (yamle.utils.trainer_utils.LoggingCallback method) on_validation_batch_end() (yamle.methods.method.BaseMethod method) (yamle.utils.trainer_utils.LoggingCallback method) on_validation_epoch_end() (yamle.methods.method.BaseMethod method) (yamle.utils.trainer_utils.LoggingCallback method) (yamle.utils.trainer_utils.ValidationModelSavingCallback method) (yamle.utils.trainer_utils.ValidationReporterMonitorCallback method) on_validation_epoch_start() (yamle.methods.method.BaseMethod method) (yamle.methods.swag.SWAGMethod method) (yamle.utils.trainer_utils.LoggingCallback method) one_hot() (in module yamle.losses.classification) one_hot_target (yamle.losses.classification.CrossEntropyLoss parameter) optimizer (yamle.methods.method.BaseMethod parameter) optimizer_id (yamle.utils.optimization_utils.set_optimizer_id parameter) optimizer_step() (yamle.methods.method.BaseMethod method) out_channels (yamle.models.operations.CompletelySeparableConv2d parameter) (yamle.models.operations.Conv2dNormActivation parameter) (yamle.models.operations.DepthwiseSeparableConv2d parameter) (yamle.models.operations.DoubleConv2d parameter) (yamle.models.unet.DownBlock parameter) (yamle.models.unet.UpBlock parameter) out_features (yamle.models.operations.LinearNormActivation parameter) outplanes (yamle.models.densenet.Transition parameter) output (yamle.utils.tracing_utils.forward_shape_hook parameter) output_activation() (in module yamle.models.operations) output_channels (yamle.models.operations.Conv2dExtractor parameter) output_kwargs (yamle.models.convnet.ConvNetModel.forward parameter) (yamle.models.densenet.DenseNetModel.forward parameter) (yamle.models.fc.FCModel.forward parameter) (yamle.models.gp.GPModel.forward parameter) (yamle.models.mixer.MixerModel.forward parameter) (yamle.models.resnet.ResNetModel.forward parameter) (yamle.models.rnn.RNNAutoEncoderModel.forward parameter) (yamle.models.rnn.RNNModel.forward parameter) (yamle.models.transformer.TransformerModel.forward parameter) (yamle.models.unet.UNetModel.forward parameter) (yamle.models.vgg.VGGModel.forward parameter) (yamle.models.visual_transformer.VisualTransformerModel.forward parameter)

OutputActivation (class in yamle.models.operations) outputs_dim (yamle.data.classification.AdultIncomeUCIClassificationDataModule attribute) (yamle.data.classification.BloodMNISTClassificationDataModule attribute) (yamle.data.classification.BreastCancerUCIClassificationDataModule attribute) (yamle.data.classification.BreastMNISTClassificationDataModule attribute) (yamle.data.classification.CarEvaluationUCIClassificationDataModule attribute) (yamle.data.classification.CreditUCIClassificationDataModule attribute) (yamle.data.classification.DermaMNISTClassificationDataModule attribute) (yamle.data.classification.DermatologyUCIClassificationDataModule attribute) (yamle.data.classification.ECG5000ClassificationDataModule attribute) (yamle.data.classification.PneumoniaMNISTClassificationDataModule attribute) (yamle.data.classification.TinyImageNetClassificationDataModule attribute) (yamle.data.classification.TorchvisionClassificationDataModuleCIFAR10 attribute) (yamle.data.classification.TorchvisionClassificationDataModuleCIFAR100 attribute) (yamle.data.classification.TorchvisionClassificationDataModuleCIFAR3 attribute) (yamle.data.classification.TorchvisionClassificationDataModuleCIFAR5 attribute) (yamle.data.classification.TorchvisionClassificationDataModuleFashionMNIST attribute) (yamle.data.classification.TorchvisionClassificationDataModuleMNIST attribute) (yamle.data.classification.TorchvisionClassificationDataModuleSVHN attribute) (yamle.data.classification.ToyTwoMoonsClassificationDataModule attribute) (yamle.data.datamodule.BaseDataModule attribute) (yamle.data.depth.DepthEstimationDataModule attribute) (yamle.data.depth.NYUv2DataModule attribute) (yamle.data.reconstruction.ECG5000ReconstructionDataModule attribute) (yamle.data.regression.RetinaMNISTDataModule attribute) (yamle.data.regression.TemperatureTimeSeriesDataModule attribute) (yamle.data.regression.TorchvisionRotationRegressionDataModule attribute) (yamle.data.regression.ToyRegressionDataModule attribute) (yamle.data.regression.UCIRegressionDataModule attribute) (yamle.data.regression.WikiFaceRegressionDataModule attribute) (yamle.data.segmentation.TorchvisionSegmentationDataModule attribute) (yamle.data.segmentation.TorchvisionSegmentationDataModuleCityscapes attribute) (yamle.data.text.Shakespeare attribute) (yamle.data.text.TorchtextClassificationDataModule attribute) (yamle.data.text.TorchtextClassificationModelIMDB attribute) (yamle.data.text.TorchtextClassificationModelWikiText103 attribute) (yamle.data.text.TorchtextClassificationModelWikiText2 attribute) (yamle.methods.method.BaseMethod parameter) (yamle.models.convnet.ConvNetModel parameter) (yamle.models.model.BaseModel parameter) (yamle.models.operations.LinearExtractor parameter) (yamle.models.operations.ParallelModel parameter) outputs_dtype (yamle.data.classification.AdultIncomeUCIClassificationDataModule attribute) (yamle.data.classification.BreastCancerUCIClassificationDataModule attribute) (yamle.data.classification.CarEvaluationUCIClassificationDataModule attribute) (yamle.data.classification.CreditUCIClassificationDataModule attribute) (yamle.data.classification.DermatologyUCIClassificationDataModule attribute) (yamle.data.classification.ECG5000ClassificationDataModule attribute) (yamle.data.classification.MedMNISTClassificationDataModule attribute) (yamle.data.classification.TorchvisionClassificationDataModule attribute) (yamle.data.classification.ToyTwoMoonsClassificationDataModule attribute) (yamle.data.datamodule.BaseDataModule attribute) (yamle.data.datamodule.RealWorldClassificationDataModule attribute) (yamle.data.datamodule.RealWorldDataModule attribute) (yamle.data.datamodule.RealWorldRegressionDataModule attribute) (yamle.data.depth.DepthEstimationDataModule attribute) (yamle.data.reconstruction.ECG5000ReconstructionDataModule attribute) (yamle.data.regression.MedMNISTRegressionDataModule attribute) (yamle.data.regression.TorchvisionRegressionDataModule attribute) (yamle.data.regression.TorchvisionRotationRegressionDataModule attribute) (yamle.data.regression.ToyRegressionDataModule attribute) (yamle.data.regression.UCIRegressionDataModule attribute) (yamle.data.regression.WikiFaceRegressionDataModule attribute) (yamle.data.segmentation.TorchvisionSegmentationDataModule attribute) (yamle.data.text.TorchtextClassificationDataModule attribute) (yamle.methods.method.BaseMethod parameter) overwrite (yamle.utils.file_utils.save_experiment parameter) overwrite_results (yamle.utils.file_utils.save_experiment parameter)

P

p (yamle.methods.mcdropout.MCDropBlockMethod parameter) (yamle.methods.mcdropout.MCDropConnectMethod parameter) (yamle.methods.mcdropout.MCDropoutMethod parameter) (yamle.methods.mcdropout.MCStochasticDepthMethod parameter) (yamle.methods.svi.SVIReparameterizationMethod parameter) (yamle.quantization.models.specific.mcdropout.QuantisedDropout1d parameter) (yamle.quantization.models.specific.mcdropout.QuantisedDropout2d parameter) (yamle.quantization.models.specific.mcdropout.QuantisedDropout3d parameter) pad_to_32 (yamle.data.classification.MedMNISTClassificationDataModule parameter) (yamle.data.classification.TorchvisionClassificationDataModule parameter) (yamle.data.regression.MedMNISTRegressionDataModule parameter) (yamle.data.regression.TorchvisionRotationRegressionDataModule parameter) (yamle.third_party.medmnist.MedMNISTDatasetWrapper parameter) padding (yamle.data.dataset_wrappers.InputImagePaddingDataset parameter) (yamle.models.operations.CompletelySeparableConv2d parameter) (yamle.models.operations.Conv2dNormActivation parameter) (yamle.models.operations.DepthwiseSeparableConv2d parameter) (yamle.models.operations.DoubleConv2d parameter) parallel (yamle.trainers.ensemble.EnsembleTrainer parameter) ParallelModel (class in yamle.models.operations) param (yamle.utils.pruning_utils.is_parameter_prunable parameter) parameters (yamle.utils.optimization_utils.disable_optimization parameter) (yamle.utils.optimization_utils.enable_optimization parameter) (yamle.utils.optimization_utils.freeze_weights parameter) (yamle.utils.optimization_utils.set_optimizer_id parameter) parent_parser (yamle.trainers.calibration.CalibrationTrainer.add_specific_args parameter) pareto_optimal() (in module yamle.utils.plotting_utils) parse_args() (in module yamle.utils.file_utils) parser (yamle.trainers.trainer.BaseTrainer.add_specific_args parameter) patch_size (yamle.models.mixer.MixerModel parameter) (yamle.models.visual_transformer.SpatialPositionalEmbedding parameter) (yamle.models.visual_transformer.VisualTransformerModel parameter) path (yamle.utils.export_utils.export_onnx parameter) PEMethod (class in yamle.methods.pe) pin_memory (yamle.data.datamodule.BaseDataModule parameter) pixelate() (in module yamle.third_party.imagenet_c.corruptions) pl_module (yamle.utils.trainer_utils.GradientNormMonitorCallback.on_after_backward parameter) (yamle.utils.trainer_utils.GradientValueClippingCallback.on_after_backward parameter) (yamle.utils.trainer_utils.L1L2MonitorCallback.on_train_epoch_end parameter) (yamle.utils.trainer_utils.ValidationReporterMonitorCallback.on_validation_epoch_end parameter) planes (yamle.models.resnet.BasicBlock parameter) (yamle.models.resnet.Blocks parameter) (yamle.models.resnet.Bottleneck parameter) (yamle.models.resnet.ResNetModel parameter) (yamle.models.vgg.Blocks parameter) (yamle.models.vgg.VGGBlock parameter) (yamle.models.vgg.VGGModel parameter) plasma_fractal() (in module yamle.third_party.imagenet_c.corruptions) plot() (yamle.data.classification.ECG5000ClassificationDataModule method) (yamle.data.classification.ToyTwoMoonsClassificationDataModule method) (yamle.data.classification.UCIClassificationDataModule method) (yamle.data.datamodule.BaseDataModule method) (yamle.data.datamodule.VisionDataModule method) (yamle.data.datamodule.VisionRegressionDataModule method) (yamle.data.depth.DepthEstimationDataModule method) (yamle.data.depth.NYUv2DataModule method) (yamle.data.reconstruction.ECG5000ReconstructionDataModule method) (yamle.data.regression.TimeSeriesRegressionDataModule method) (yamle.data.regression.ToyRegressionDataModule method) (yamle.data.regression.UCIRegressionDataModule method) (yamle.data.segmentation.TorchvisionSegmentationDataModule method) (yamle.data.segmentation.TorchvisionSegmentationDataModuleCityscapes method) (yamle.data.text.TorchtextClassificationDataModule method) plot_different_metrics_and_trial_id() (in module yamle.utils.tuning_utils) plot_different_runs_and_metric_config_combinations() (in module yamle.utils.tuning_utils) plot_different_runs_and_metrics() (in module yamle.utils.tuning_utils) plot_input_layer_norm_bar() (in module yamle.utils.specific.mimo_experiments.plotting_utils) plot_output_layer_norm_bar() (in module yamle.utils.specific.mimo_experiments.plotting_utils) plot_overlap_between_members() (in module yamle.utils.specific.mimo_experiments.plotting_utils) plot_weight_histogram() (in module yamle.utils.specific.mimo_experiments.plotting_utils) plot_weight_trajectories() (in module yamle.utils.specific.mimo_experiments.plotting_utils) plots_file() (in module yamle.utils.file_utils) plotting_testing (yamle.methods.method.BaseMethod parameter) plotting_training (yamle.methods.method.BaseMethod parameter) PneumoniaMNISTClassificationDataModule (class in yamle.data.classification) pool (yamle.models.operations.Conv2dExtractor parameter) pool_kwargs (yamle.models.operations.Pooling parameter) Pooling (class in yamle.models.operations) pooling (yamle.models.operations.Conv2dExtractor parameter) (yamle.models.operations.Pooling parameter) (yamle.models.vgg.Blocks parameter) (yamle.models.vgg.VGGBlock parameter) (yamle.models.vgg.VGGModel parameter) (yamle.models.visual_transformer.VisualTransformerModel parameter) positional_embedding (yamle.models.visual_transformer.SpatialPositionalEmbedding parameter) PositionalEncoding (class in yamle.models.transformer) PowerGrowthScalarScheduler (class in yamle.utils.optimization_utils) precision (yamle.trainers.trainer.BaseTrainer parameter) predictions (yamle.utils.operation_utils.average_predictions parameter) (yamle.utils.operation_utils.weighted_regression_variance parameter)

predictions_file() (in module yamle.utils.file_utils) prefix (yamle.methods.method.BaseMethod.reset_metrics parameter) (yamle.utils.file_utils.store_metrics parameter) (yamle.utils.running_utils.extract_kwargs parameter) PreNorm (class in yamle.models.transformer) prepare() (yamle.quantization.qat.QATQuantizer method) (yamle.quantization.quantizer.BaseQuantizer method) (yamle.quantization.quantizer.DummyQuantizer method) (yamle.quantization.static.StaticQuantizer method) prepare_data() (yamle.data.classification.ECG5000ClassificationDataModule method) (yamle.data.classification.MedMNISTClassificationDataModule method) (yamle.data.classification.TorchvisionClassificationDataModule method) (yamle.data.classification.TorchvisionClassificationDataModuleFashionMNIST method) (yamle.data.classification.TorchvisionClassificationDataModuleMNIST method) (yamle.data.classification.ToyTwoCirclesClassificationDataModule method) (yamle.data.classification.ToyTwoMoonsClassificationDataModule method) (yamle.data.custom.ECG5000DataModule method) (yamle.data.datamodule.BaseDataModule method) (yamle.data.datamodule.RealWorldDataModule method) (yamle.data.datamodule.RealWorldRegressionDataModule method) (yamle.data.depth.DepthEstimationDataModule method) (yamle.data.reconstruction.ECG5000ReconstructionDataModule method) (yamle.data.regression.MedMNISTRegressionDataModule method) (yamle.data.regression.TelemonitoringUCIRegressionDataModule method) (yamle.data.regression.TimeSeriesRegressionDataModule method) (yamle.data.regression.TorchvisionRegressionDataModule method) (yamle.data.regression.TorchvisionRotationRegressionDataModule method) (yamle.data.regression.TorchvisionRotationRegressionDataModuleFashionMNIST method) (yamle.data.regression.TorchvisionRotationRegressionDataModuleMNIST method) (yamle.data.regression.ToyRegressionDataModule method) (yamle.data.segmentation.TorchvisionSegmentationDataModule method) (yamle.data.text.TorchtextClassificationDataModule method) prepare_data_per_node (yamle.methods.augmentation_classification.CutMixImageClassificationMethod attribute) (yamle.methods.augmentation_classification.CutOutImageClassificationMethod attribute) (yamle.methods.augmentation_classification.MixUpImageClassificationMethod attribute) (yamle.methods.augmentation_classification.RandomErasingImageClassificationMethod attribute) (yamle.methods.mimo.DataMUXMethod attribute) (yamle.methods.mimo.MIMMOMethod attribute) (yamle.methods.mimo.MIMOMethod attribute) (yamle.methods.mimo.MixMoMethod attribute) (yamle.methods.mimo.MixVitMethod attribute) (yamle.methods.mimo.UnMixMoMethod attribute) (yamle.methods.moe.MixtureOfExpertsMethod attribute) (yamle.methods.moe.MultiHeadEnsembleMethod attribute) (yamle.methods.pe.PEMethod attribute) (yamle.methods.rbnn.RBNNMethod attribute) (yamle.methods.sgld.SGLDMethod attribute) (yamle.methods.sngp.SNGPMethod attribute) (yamle.methods.svi.SVIFlipOutDropConnectMethod attribute) (yamle.methods.svi.SVIFlipOutRTMethod attribute) (yamle.methods.svi.SVILRTMethod attribute) (yamle.methods.svi.SVILRTVDMethod attribute) (yamle.methods.svi.SVIReparameterizationMethod attribute) (yamle.methods.svi.SVIRTMethod attribute) (yamle.methods.swag.SWAGMethod attribute) (yamle.methods.temperature_scaling.TemperatureMethod attribute) (yamle.methods.uncertain_method.MCSamplingMethod attribute) (yamle.methods.uncertain_method.MemberMethod attribute) (yamle.methods.uncertain_method.SVIMethod attribute) prepare_datamodule_kwargs() (in module yamle.utils.running_utils) prepare_loss_kwargs() (in module yamle.utils.running_utils) prepare_method_kwargs() (in module yamle.utils.running_utils) prepare_metrics_kwargs() (in module yamle.utils.running_utils) prepare_model_kwargs() (in module yamle.utils.running_utils) prepare_pruner_kwargs() (in module yamle.utils.running_utils) prepare_quantizer_kwargs() (in module yamle.utils.running_utils) prepare_regularizer_kwargs() (in module yamle.utils.running_utils) prepare_test_trainer_kwargs() (in module yamle.utils.running_utils) prepare_trainer_kwargs() (in module yamle.utils.running_utils) prior (yamle.methods.mimo.MIMMOMethod parameter) prior_betas (yamle.methods.dun.DUNMethod property) prior_covariance (yamle.methods.gp.GPMethod parameter) (yamle.models.gp.GPModel parameter) prior_log_variance (yamle.methods.rbnn.RBNNMethod parameter) (yamle.methods.rbnn.replace_with_rbnn parameter) (yamle.methods.svi.SVIReparameterizationMethod parameter) prior_mean (yamle.methods.gp.GPMethod parameter) (yamle.methods.rbnn.RBNNMethod parameter) (yamle.methods.rbnn.replace_with_rbnn parameter) (yamle.methods.svi.SVIReparameterizationMethod parameter) (yamle.models.gp.GPModel parameter) probabilities (yamle.utils.operation_utils.classification_uncertainty_decomposition parameter) (yamle.utils.operation_utils.entropy parameter) profiler (yamle.trainers.trainer.BaseTrainer parameter) prune_parameter() (yamle.pruning.pruner.BasePruner static method) pruner_factory() (in module yamle.pruning) pruner_type (yamle.pruning.pruner_factory parameter) pruning_percentage (yamle.pruning.unstructured.magnitude.UnstructuredMagnitudePruner parameter)

Q

QATQuantizer (class in yamle.quantization.qat) QuantileRegressionLoss (class in yamle.losses.regression) quantiles (yamle.losses.regression.QuantileRegressionLoss parameter) QuantisedDropout1d (class in yamle.quantization.models.specific.mcdropout)

QuantisedDropout2d (class in yamle.quantization.models.specific.mcdropout) QuantisedDropout3d (class in yamle.quantization.models.specific.mcdropout) QuantizableAdd (class in yamle.quantization.models.operations) quantizer_factory() (in module yamle.quantization) quantizer_type (yamle.quantization.quantizer_factory parameter)

R

r (yamle.methods.mimo.MixMoMethod parameter) rand_bbox() (in module yamle.data.augmentations) random_augment_magnitude (yamle.methods.contrastive.SimCLRVisionMethod parameter) random_augment_num_ops (yamle.methods.contrastive.SimCLRVisionMethod parameter) random_erasing_ratio (yamle.methods.augmentation_classification.RandomErasingImageClassificationMethod parameter) random_erasing_scale (yamle.methods.augmentation_classification.RandomErasingImageClassificationMethod parameter) random_erasing_value (yamle.methods.augmentation_classification.RandomErasingImageClassificationMethod parameter) random_features (yamle.methods.sngp.SNGPMethod parameter) RandomErasing (class in yamle.data.augmentations) RandomErasingImageClassificationMethod (class in yamle.methods.augmentation_classification) RandomImageNoise (class in yamle.third_party.imagenet_c.extra) RandomTabularNoise (class in yamle.third_party.imagenet_c.extra) ratio (yamle.data.augmentations.RandomErasing parameter) RBNNMethod (class in yamle.methods.rbnn) RealWorldClassificationDataModule (class in yamle.data.datamodule) RealWorldDataModule (class in yamle.data.datamodule) RealWorldRegressionDataModule (class in yamle.data.datamodule) recover() (yamle.pruning.pruner.BasePruner method) (yamle.quantization.quantizer.BaseQuantizer method) recover_frozen_weights() (in module yamle.utils.optimization_utils) recover_parameter() (yamle.pruning.pruner.BasePruner static method) Reduction (class in yamle.models.operations) reduction (yamle.models.operations.Reduction parameter) (yamle.models.operations.SqueezeAndExcitation parameter) reduction_per_feature (yamle.losses.loss.BaseLoss parameter) reduction_per_member (yamle.losses.loss.BaseLoss parameter) reduction_per_sample (yamle.losses.loss.BaseLoss parameter) regression_uncertainty_decomposition() (in module yamle.utils.operation_utils) RegularizedWeightsMonitorCallback (class in yamle.utils.trainer_utils) regularizer (yamle.methods.method.BaseMethod parameter) regularizer_factory() (in module yamle.regularizers) regularizer_type (yamle.regularizers.regularizer_factory parameter) regularizer_weight (yamle.methods.method.BaseMethod parameter) remove_argparse_argument() (in module yamle.utils.file_utils) repeat_evaluation (yamle.methods.mimo.MIMOMethod parameter) repeat_inputs() (in module yamle.utils.operation_utils) repeat_inputs_in_batch() (in module yamle.utils.operation_utils) replace_arguments (yamle.utils.file_utils.argparse_to_config_file parameter) replace_layers_for_quantization() (yamle.models.densenet.DenseLayer method) (yamle.models.densenet.DenseNetModel method) (yamle.models.fc.FCModel method) (yamle.models.model.BaseModel method) (yamle.models.operations.Conv2dNormActivation method) (yamle.models.operations.LinearNormActivation method) (yamle.models.resnet.BasicBlock method) (yamle.models.resnet.Bottleneck method) (yamle.models.resnet.ResNetModel method) (yamle.models.vgg.VGGBlock method) (yamle.models.vgg.VGGModel method) (yamle.quantization.quantizer.BaseQuantizer method)

replace_layers_with_grouped_convs() (in module yamle.methods.pe) replace_with_be() (in module yamle.methods.be) replace_with_rbnn() (in module yamle.methods.rbnn) reporter (yamle.utils.trainer_utils.ValidationReporterMonitorCallback parameter) reset() (yamle.models.gp.GPModel method) (yamle.models.model.BaseModel method) (yamle.utils.optimization_utils.CosineScalarScheduler method) (yamle.utils.optimization_utils.LinearScalarScheduler method) (yamle.utils.optimization_utils.PowerGrowthScalarScheduler method) (yamle.utils.optimization_utils.ScalarScheduler method) (yamle.utils.optimization_utils.SineScalarScheduler method) reset_metrics() (yamle.methods.method.BaseMethod method) reset_parameters() (yamle.models.operations.CompletelySeparableConv2d method) (yamle.models.operations.DepthwiseSeparableConv2d method) (yamle.models.transformer.PositionalEncoding method) (yamle.models.transformer.TransformerModel method) ReshapeInput (class in yamle.models.operations) ReshapeOutput (class in yamle.models.operations) residual (yamle.models.operations.Conv2dExtractor parameter) (yamle.models.operations.DoubleConv2d parameter) (yamle.models.operations.LinearExtractor parameter) ResidualConvNetModel (class in yamle.models.convnet) ResidualFCModel (class in yamle.models.fc) ResidualLayer (class in yamle.models.operations) ResNetModel (class in yamle.models.resnet) result (yamle.utils.plotting_utils.average_augmentations parameter) results (yamle.utils.file_utils.save_experiment parameter) (yamle.utils.file_utils.store_metrics parameter) results_file() (in module yamle.utils.file_utils) RetinaMNISTDataModule (class in yamle.data.regression) RNNAutoEncoderModel (class in yamle.models.rnn) RNNModel (class in yamle.models.rnn) root (yamle.third_party.nyuv2.NYUv2 parameter) root_dir (yamle.data.regression.WikiFaceDataset parameter)

S

sample_data() (yamle.data.datamodule.BaseDataModule method) sample_initial_random_configs() (in module yamle.utils.tuning_utils) samples (yamle.utils.operation_utils.classification_uncertainty_decomposition parameter) (yamle.utils.operation_utils.regression_uncertainty_decomposition parameter) sampling_epochs (yamle.methods.sgld.SGLDMethod parameter) saturate() (in module yamle.third_party.imagenet_c.corruptions) save_args() (in module yamle.utils.file_utils) save_args_dictionary() (in module yamle.utils.file_utils) save_experiment() (in module yamle.utils.file_utils) save_method() (in module yamle.utils.file_utils) save_model() (in module yamle.utils.file_utils) save_model_onnx() (in module yamle.utils.file_utils) save_original_model() (yamle.quantization.quantizer.BaseQuantizer method) save_path (yamle.data.regression.TimeSeriesRegressionDataModule.plot parameter) (yamle.methods.method.BaseMethod parameter) (yamle.trainers.trainer.BaseTrainer parameter) (yamle.utils.file_utils.save_experiment parameter) save_pickle() (in module yamle.utils.file_utils) save_quantized_model() (yamle.quantization.quantizer.BaseQuantizer method) save_results() (in module yamle.utils.file_utils) save_scripts() (in module yamle.utils.file_utils) save_test_predictions (yamle.methods.method.BaseMethod parameter) save_tuning_results() (in module yamle.utils.file_utils) ScalarAdder (class in yamle.models.operations) ScalarMultiplier (class in yamle.models.operations) ScalarScheduler (class in yamle.utils.optimization_utils) scale (yamle.data.augmentations.RandomErasing parameter) (yamle.data.transforms.TargetToUnit parameter) (yamle.methods.swag.SWAGMethod parameter) scheduler (yamle.methods.method.BaseMethod parameter) scheduler_factor (yamle.methods.method.BaseMethod parameter) scheduler_gamma (yamle.methods.method.BaseMethod parameter) scheduler_patience (yamle.methods.method.BaseMethod parameter) scheduler_step_size (yamle.methods.method.BaseMethod parameter) se (yamle.models.operations.Conv2dExtractor parameter) seed (yamle.data.datamodule.BaseDataModule parameter) (yamle.data.datamodule.VisionDataModule parameter) (yamle.data.dataset_wrappers.ImageRotationDataset parameter) (yamle.data.regression.WikiFaceDataset parameter) (yamle.data.segmentation.TorchvisionSegmentationDataModule parameter) (yamle.data.text.TorchtextClassificationDataModule parameter) (yamle.methods.method.BaseMethod parameter) set_hard_value() (yamle.utils.optimization_utils.CosineScalarScheduler method) (yamle.utils.optimization_utils.LinearScalarScheduler method) (yamle.utils.optimization_utils.PowerGrowthScalarScheduler method) (yamle.utils.optimization_utils.SineScalarScheduler method) set_optimizer_id() (in module yamle.utils.optimization_utils) set_reduction_per_feature() (yamle.losses.loss.BaseLoss method) set_reduction_per_member() (yamle.losses.loss.BaseLoss method) set_reduction_per_sample() (yamle.losses.loss.BaseLoss method) setup() (yamle.data.custom.ECG5000DataModule method) (yamle.data.datamodule.BaseDataModule method) (yamle.data.datamodule.RealWorldDataModule method) (yamle.data.datamodule.VisionDataModule method) severity (yamle.third_party.imagenet_c.tabular_corruptions.additive_gaussian_noise parameter) (yamle.third_party.imagenet_c.TabularCorruption parameter) (yamle.third_party.imagenet_c.VisionCorruption parameter) SGLDMethod (class in yamle.methods.sgld) Shakespeare (class in yamle.data.text) shape (yamle.models.operations.ScalarAdder parameter) (yamle.models.operations.ScalarMultiplier parameter) shot_noise() (in module yamle.third_party.imagenet_c.corruptions) shrinkage (yamle.methods.ensemble.GradientBoostingEnsembleMethod parameter) ShrinkAndPerturbRegularizer (class in yamle.regularizers.model) SimCLRVisionMethod (class in yamle.methods.contrastive) similarity (yamle.losses.contrastive.NoiseContrastiveEstimatorLoss parameter) SineScalarScheduler (class in yamle.utils.optimization_utils) single_source (yamle.models.operations.ParallelModel parameter) size (yamle.data.augmentations.CutOut parameter) (yamle.data.augmentations.rand_bbox parameter) (yamle.third_party.imagenet_c.extra.RandomImageNoise parameter) (yamle.third_party.imagenet_c.extra.RandomTabularNoise parameter) SnapsotEnsembleMethod (class in yamle.methods.ensemble) SNGPMethod (class in yamle.methods.sngp) snow() (in module yamle.third_party.imagenet_c.corruptions) SoftIntersectionOverUnionLoss (class in yamle.losses.segmentation) SpatialPositionalEmbedding (class in yamle.models.visual_transformer) spatter() (in module yamle.third_party.imagenet_c.corruptions) specific_name (yamle.data.regression.TimeSeriesRegressionDataModule.plot parameter) speckle_noise() (in module yamle.third_party.imagenet_c.corruptions) split (yamle.data.regression.WikiFaceDataset parameter) split_folder (yamle.third_party.tinyimagenet.TinyImageNet property) split_optimizer_parameters() (in module yamle.utils.optimization_utils) SplitParametersMonitorCallback (class in yamle.utils.trainer_utils) splits (yamle.third_party.tinyimagenet.TinyImageNet attribute) SqueezeAndExcitation (class in yamle.models.operations) st_checkpoint_dir (yamle.trainers.trainer.BaseTrainer parameter) staged_output (yamle.models.convnet.ConvNetModel.forward parameter) (yamle.models.densenet.DenseNetModel.forward parameter) (yamle.models.fc.FCModel.forward parameter) (yamle.models.gp.GPModel.forward parameter) (yamle.models.mixer.MixerModel.forward parameter) (yamle.models.resnet.ResNetModel.forward parameter) (yamle.models.rnn.RNNAutoEncoderModel.forward parameter) (yamle.models.rnn.RNNModel.forward parameter) (yamle.models.transformer.TransformerModel.forward parameter) (yamle.models.unet.UNetModel.forward parameter) (yamle.models.vgg.VGGModel.forward parameter) (yamle.models.visual_transformer.VisualTransformerModel.forward parameter)

start_epoch (yamle.regularizers.model.ShrinkAndPerturbRegularizer parameter) (yamle.utils.optimization_utils.CosineScalarScheduler parameter) (yamle.utils.optimization_utils.LinearScalarScheduler parameter) (yamle.utils.optimization_utils.PowerGrowthScalarScheduler parameter) (yamle.utils.optimization_utils.SineScalarScheduler parameter) start_value (yamle.utils.optimization_utils.CosineScalarScheduler parameter) (yamle.utils.optimization_utils.LinearScalarScheduler parameter) (yamle.utils.optimization_utils.PowerGrowthScalarScheduler parameter) (yamle.utils.optimization_utils.SineScalarScheduler parameter) state_dict() (yamle.methods.delta_uq.DeltaUQMethod method) (yamle.methods.dun.DUNMethod method) (yamle.methods.gp.GPMethod method) (yamle.methods.laplace.LaplaceMethod method) (yamle.methods.method.BaseMethod method) (yamle.methods.mimo.MIMMOMethod method) (yamle.methods.mimo.MIMOMethod method) (yamle.methods.mimo.UnMixMoMethod method) (yamle.methods.swag.SWAGMethod method) (yamle.utils.optimization_utils.CosineScalarScheduler method) (yamle.utils.optimization_utils.LinearScalarScheduler method) (yamle.utils.optimization_utils.PowerGrowthScalarScheduler method) (yamle.utils.optimization_utils.ScalarScheduler method) (yamle.utils.optimization_utils.SineScalarScheduler method) StaticQuantizer (class in yamle.quantization.static) std (yamle.data.classification.BloodMNISTClassificationDataModule attribute) (yamle.data.classification.BreastMNISTClassificationDataModule attribute) (yamle.data.classification.DermaMNISTClassificationDataModule attribute) (yamle.data.classification.PneumoniaMNISTClassificationDataModule attribute) (yamle.data.classification.TinyImageNetClassificationDataModule attribute) (yamle.data.classification.TorchvisionClassificationDataModuleCIFAR10 attribute) (yamle.data.classification.TorchvisionClassificationDataModuleCIFAR100 attribute) (yamle.data.classification.TorchvisionClassificationDataModuleCIFAR3 attribute) (yamle.data.classification.TorchvisionClassificationDataModuleCIFAR5 attribute) (yamle.data.classification.TorchvisionClassificationDataModuleFashionMNIST attribute) (yamle.data.classification.TorchvisionClassificationDataModuleMNIST attribute) (yamle.data.classification.TorchvisionClassificationDataModuleSVHN attribute) (yamle.data.datamodule.VisionDataModule attribute) (yamle.data.depth.DepthEstimationDataModule attribute) (yamle.data.depth.NYUv2DataModule attribute) (yamle.data.regression.RetinaMNISTDataModule attribute) (yamle.data.regression.TinyImageNetRotationRegressionDataModule attribute) (yamle.data.regression.TorchvisionRotationRegressionDataModuleCIFAR10 attribute) (yamle.data.regression.TorchvisionRotationRegressionDataModuleCIFAR100 attribute) (yamle.data.regression.TorchvisionRotationRegressionDataModuleFashionMNIST attribute) (yamle.data.regression.TorchvisionRotationRegressionDataModuleMNIST attribute) (yamle.data.regression.TorchvisionRotationRegressionDataModuleSVHN attribute) (yamle.data.regression.WikiFaceRegressionDataModule attribute) (yamle.data.segmentation.TorchvisionSegmentationDataModule attribute) (yamle.data.segmentation.TorchvisionSegmentationDataModuleCityscapes attribute) (yamle.data.text.TorchtextClassificationDataModule attribute) (yamle.data.transforms.Denormalize parameter) (yamle.data.transforms.JointNormalize parameter) (yamle.data.transforms.Normalize parameter) (yamle.regularizers.model.ShrinkAndPerturbRegularizer parameter) (yamle.third_party.imagenet_c.extra.RandomImageNoise parameter) (yamle.third_party.imagenet_c.extra.RandomTabularNoise parameter) step() (yamle.utils.optimization_utils.CosineScalarScheduler method) (yamle.utils.optimization_utils.LinearScalarScheduler method) (yamle.utils.optimization_utils.PowerGrowthScalarScheduler method) (yamle.utils.optimization_utils.ScalarScheduler method) (yamle.utils.optimization_utils.SineScalarScheduler method) store_metrics() (in module yamle.utils.file_utils) stride (yamle.models.operations.CompletelySeparableConv2d parameter) (yamle.models.operations.Conv2dNormActivation parameter) (yamle.models.operations.DepthwiseSeparableConv2d parameter) (yamle.models.operations.DoubleConv2d parameter) (yamle.models.resnet.BasicBlock parameter) (yamle.models.resnet.Blocks parameter) (yamle.models.resnet.Bottleneck parameter) strides (yamle.models.resnet.ResNetModel parameter) summary() (yamle.pruning.pruner.BasePruner method) (yamle.pruning.pruner.DummyPruner method) SurrogateDataset (class in yamle.data.dataset_wrappers) SVIFlipOutDropConnectMethod (class in yamle.methods.svi) SVIFlipOutRTMethod (class in yamle.methods.svi) SVILRTMethod (class in yamle.methods.svi) SVILRTVDMethod (class in yamle.methods.svi) SVIMethod (class in yamle.methods.uncertain_method) SVIReparameterizationMethod (class in yamle.methods.svi) SVIRTMethod (class in yamle.methods.svi) SWAGMethod (class in yamle.methods.swag)

T

T (yamle.models.rnn.InputRepeater.forward parameter) TabularCorruption (class in yamle.third_party.imagenet_c) target_dtype (yamle.data.transforms.JointToTensor parameter) target_normalization (yamle.third_party.medmnist.MedMNISTDatasetWrapper parameter) target_transform (yamle.data.dataset_wrappers.SurrogateDataset parameter) targets (yamle.data.augmentations.CutMix.augment parameter) (yamle.data.augmentations.CutOut.augment parameter) (yamle.data.augmentations.RandomErasing.augment parameter) targets_dim (yamle.data.classification.BloodMNISTClassificationDataModule attribute) (yamle.data.classification.BreastMNISTClassificationDataModule attribute) (yamle.data.classification.DermaMNISTClassificationDataModule attribute) (yamle.data.classification.ECG5000ClassificationDataModule attribute) (yamle.data.classification.PneumoniaMNISTClassificationDataModule attribute) (yamle.data.classification.TinyImageNetClassificationDataModule attribute) (yamle.data.classification.TorchvisionClassificationDataModuleCIFAR10 attribute) (yamle.data.classification.TorchvisionClassificationDataModuleCIFAR100 attribute) (yamle.data.classification.TorchvisionClassificationDataModuleCIFAR3 attribute) (yamle.data.classification.TorchvisionClassificationDataModuleCIFAR5 attribute) (yamle.data.classification.TorchvisionClassificationDataModuleFashionMNIST attribute) (yamle.data.classification.TorchvisionClassificationDataModuleMNIST attribute) (yamle.data.classification.TorchvisionClassificationDataModuleSVHN attribute) (yamle.data.classification.ToyTwoMoonsClassificationDataModule attribute) (yamle.data.datamodule.BaseDataModule attribute) (yamle.data.datamodule.RealWorldClassificationDataModule attribute) (yamle.data.datamodule.RealWorldDataModule attribute) (yamle.data.datamodule.RealWorldRegressionDataModule attribute) (yamle.data.depth.NYUv2DataModule attribute) (yamle.data.reconstruction.ECG5000ReconstructionDataModule attribute) (yamle.data.regression.RetinaMNISTDataModule attribute) (yamle.data.regression.TorchvisionRotationRegressionDataModule attribute) (yamle.data.regression.ToyRegressionDataModule attribute) (yamle.data.regression.WikiFaceRegressionDataModule attribute) (yamle.data.segmentation.TorchvisionSegmentationDataModuleCityscapes attribute) (yamle.data.text.Shakespeare attribute) (yamle.data.text.TorchtextClassificationModelIMDB attribute) (yamle.data.text.TorchtextClassificationModelWikiText103 attribute) (yamle.data.text.TorchtextClassificationModelWikiText2 attribute) (yamle.methods.method.BaseMethod parameter) TargetToUnit (class in yamle.data.transforms) task (yamle.data.classification.AdultIncomeUCIClassificationDataModule attribute) (yamle.data.classification.BreastCancerUCIClassificationDataModule attribute) (yamle.data.classification.CarEvaluationUCIClassificationDataModule attribute) (yamle.data.classification.CreditUCIClassificationDataModule attribute) (yamle.data.classification.DermatologyUCIClassificationDataModule attribute) (yamle.data.classification.ECG5000ClassificationDataModule attribute) (yamle.data.classification.ToyTwoMoonsClassificationDataModule attribute) (yamle.data.datamodule.BaseDataModule attribute) (yamle.data.datamodule.VisionClassificationDataModule attribute) (yamle.data.datamodule.VisionRegressionDataModule attribute) (yamle.data.depth.DepthEstimationDataModule attribute) (yamle.data.reconstruction.ECG5000ReconstructionDataModule attribute) (yamle.data.regression.TemperatureTimeSeriesDataModule attribute) (yamle.data.regression.ToyRegressionDataModule attribute) (yamle.data.regression.UCIRegressionDataModule attribute) (yamle.data.segmentation.TorchvisionSegmentationDataModule attribute) (yamle.data.text.TorchtextClassificationDataModule attribute) (yamle.methods.method.BaseMethod parameter) (yamle.models.convnet.ConvNetModel parameter) (yamle.models.gp.GPModel parameter) (yamle.models.mixer.MixerModel parameter) (yamle.models.model.BaseModel parameter) (yamle.models.operations.output_activation parameter) (yamle.models.operations.OutputActivation parameter) (yamle.third_party.nyuv2.NYUv2 parameter) (yamle.trainers.trainer.BaseTrainer parameter) (yamle.utils.operation_utils.average_predictions parameter) (yamle.utils.trainer_utils.ValidationModelSavingCallback parameter) tasks (yamle.losses.classification.CrossEntropyLoss attribute) (yamle.losses.classification.TextCrossEntropyLoss attribute) (yamle.losses.evidential_regression.EvidentialRegressionLoss attribute) (yamle.losses.loss.BaseLoss attribute) (yamle.losses.regression.GaussianNegativeLogLikelihoodLoss attribute) (yamle.losses.regression.MeanSquaredError attribute) (yamle.losses.regression.QuantileRegressionLoss attribute) (yamle.methods.augmentation_classification.AugmentationImageClassificationMethod attribute) (yamle.methods.evidential_regression.EvidentialRegressionMethod attribute) (yamle.methods.gp.GPMethod attribute) (yamle.methods.laplace.LaplaceMethod attribute) (yamle.methods.method.BaseMethod attribute) (yamle.methods.sngp.SNGPMethod attribute) (yamle.methods.temperature_scaling.TemperatureMethod attribute) (yamle.models.convnet.ConvNetModel attribute) (yamle.models.densenet.DenseNetModel attribute) (yamle.models.fc.FCModel attribute) (yamle.models.mixer.MixerModel attribute) (yamle.models.model.BaseModel attribute) (yamle.models.resnet.ResNetModel attribute) (yamle.models.rnn.RNNAutoEncoderModel attribute) (yamle.models.rnn.RNNModel attribute) (yamle.models.transformer.TransformerModel attribute) (yamle.models.unet.UNetModel attribute) (yamle.models.vgg.VGGModel attribute) (yamle.models.visual_transformer.VisualTransformerModel attribute) TelemonitoringUCIRegressionDataModule (class in yamle.data.regression) temperature (yamle.losses.contrastive.NoiseContrastiveEstimatorLoss parameter) TemperatureMethod (class in yamle.methods.temperature_scaling) TemperatureTimeSeriesDataModule (class in yamle.data.regression) test() (yamle.trainers.trainer.BaseTrainer method) test_augmentation() (yamle.data.datamodule.BaseDataModule method) test_augmentations (yamle.data.classification.BloodMNISTClassificationDataModule attribute) (yamle.data.classification.BreastMNISTClassificationDataModule attribute) (yamle.data.classification.DermaMNISTClassificationDataModule attribute) (yamle.data.classification.PneumoniaMNISTClassificationDataModule attribute) (yamle.data.classification.TinyImageNetClassificationDataModule attribute) (yamle.data.classification.TorchvisionClassificationDataModuleCIFAR10 attribute) (yamle.data.classification.TorchvisionClassificationDataModuleCIFAR100 attribute) (yamle.data.classification.TorchvisionClassificationDataModuleCIFAR3 attribute) (yamle.data.classification.TorchvisionClassificationDataModuleCIFAR5 attribute) (yamle.data.classification.TorchvisionClassificationDataModuleFashionMNIST attribute) (yamle.data.classification.TorchvisionClassificationDataModuleMNIST attribute) (yamle.data.classification.TorchvisionClassificationDataModuleSVHN attribute) (yamle.data.classification.ToyTwoCirclesClassificationDataModule attribute) (yamle.data.datamodule.BaseDataModule parameter) (yamle.data.datamodule.RealWorldClassificationDataModule attribute) (yamle.data.datamodule.RealWorldDataModule attribute) (yamle.data.datamodule.RealWorldRegressionDataModule attribute) (yamle.data.datamodule.VisionClassificationDataModule attribute) (yamle.data.datamodule.VisionRegressionDataModule attribute) (yamle.data.depth.DepthEstimationDataModule attribute) (yamle.data.depth.NYUv2DataModule attribute) (yamle.data.reconstruction.ECG5000ReconstructionDataModule attribute) (yamle.data.regression.AbaloneUCIRegressionDataModule attribute) (yamle.data.regression.BostonUCIRegressionDataModule attribute) (yamle.data.regression.ConcreteUCIRegressionDataModule attribute) (yamle.data.regression.EnergyUCIRegressionDataModule attribute) (yamle.data.regression.MedMNISTRegressionDataModule attribute) (yamle.data.regression.RetinaMNISTDataModule attribute) (yamle.data.regression.TelemonitoringUCIRegressionDataModule attribute) (yamle.data.regression.TemperatureTimeSeriesDataModule attribute) (yamle.data.regression.TimeSeriesRegressionDataModule attribute) (yamle.data.regression.TinyImageNetRotationRegressionDataModule attribute) (yamle.data.regression.TorchvisionRegressionDataModule attribute) (yamle.data.regression.TorchvisionRotationRegressionDataModule attribute) (yamle.data.regression.TorchvisionRotationRegressionDataModuleCIFAR10 attribute) (yamle.data.regression.TorchvisionRotationRegressionDataModuleCIFAR100 attribute) (yamle.data.regression.TorchvisionRotationRegressionDataModuleFashionMNIST attribute) (yamle.data.regression.TorchvisionRotationRegressionDataModuleMNIST attribute) (yamle.data.regression.TorchvisionRotationRegressionDataModuleSVHN attribute) (yamle.data.regression.ToyRegressionDataModule attribute) (yamle.data.regression.UCIRegressionDataModule attribute) (yamle.data.regression.WikiFaceRegressionDataModule attribute) (yamle.data.regression.WineQualityUCIRegressionDataModule attribute) (yamle.data.regression.YachtUCIRegressionDataModule attribute) (yamle.data.segmentation.TorchvisionSegmentationDataModule attribute) (yamle.data.segmentation.TorchvisionSegmentationDataModuleCityscapes attribute) (yamle.data.text.Shakespeare attribute) (yamle.data.text.TorchtextClassificationDataModule attribute) (yamle.data.text.TorchtextClassificationModelIMDB attribute) (yamle.data.text.TorchtextClassificationModelWikiText103 attribute) (yamle.data.text.TorchtextClassificationModelWikiText2 attribute) test_dataloader() (yamle.data.datamodule.BaseDataModule method) test_dataset() (yamle.data.datamodule.BaseDataModule method) test_dataset_size() (yamle.data.datamodule.BaseDataModule method) test_joint_transform (yamle.data.datamodule.BaseDataModule parameter) test_joint_transform() (yamle.data.datamodule.BaseDataModule method) test_name (yamle.methods.augmentation_classification.CutMixImageClassificationMethod attribute) (yamle.methods.augmentation_classification.CutOutImageClassificationMethod attribute) (yamle.methods.augmentation_classification.MixUpImageClassificationMethod attribute) (yamle.methods.augmentation_classification.RandomErasingImageClassificationMethod attribute) (yamle.methods.mimo.DataMUXMethod attribute) (yamle.methods.mimo.MIMMOMethod attribute) (yamle.methods.mimo.MIMOMethod attribute) (yamle.methods.mimo.MixMoMethod attribute) (yamle.methods.mimo.MixVitMethod attribute) (yamle.methods.mimo.UnMixMoMethod attribute) (yamle.methods.moe.MixtureOfExpertsMethod attribute) (yamle.methods.moe.MultiHeadEnsembleMethod attribute) (yamle.methods.pe.PEMethod attribute) (yamle.methods.rbnn.RBNNMethod attribute) (yamle.methods.sgld.SGLDMethod attribute) (yamle.methods.sngp.SNGPMethod attribute) (yamle.methods.svi.SVIFlipOutDropConnectMethod attribute) (yamle.methods.svi.SVIFlipOutRTMethod attribute) (yamle.methods.svi.SVILRTMethod attribute) (yamle.methods.svi.SVILRTVDMethod attribute) (yamle.methods.svi.SVIReparameterizationMethod attribute) (yamle.methods.svi.SVIRTMethod attribute) (yamle.methods.swag.SWAGMethod attribute) (yamle.methods.temperature_scaling.TemperatureMethod attribute) (yamle.methods.uncertain_method.MCSamplingMethod attribute) (yamle.methods.uncertain_method.MemberMethod attribute) (yamle.methods.uncertain_method.SVIMethod attribute)

test_number_of_batches() (yamle.data.datamodule.BaseDataModule method) test_portion (yamle.data.datamodule.BaseDataModule parameter) (yamle.data.datamodule.RealWorldClassificationDataModule parameter) (yamle.data.datamodule.RealWorldRegressionDataModule parameter) (yamle.data.regression.WikiFaceDataset parameter) test_samples (yamle.data.regression.ToyRegressionDataModule parameter) test_step() (yamle.methods.method.BaseMethod method) test_target_transform (yamle.data.datamodule.BaseDataModule parameter) test_target_transform() (yamle.data.datamodule.BaseDataModule method) test_transform (yamle.data.datamodule.BaseDataModule parameter) (yamle.data.datamodule.VisionDataModule parameter) (yamle.data.segmentation.TorchvisionSegmentationDataModule parameter) test_transform() (yamle.data.datamodule.BaseDataModule method) tester (yamle.data.regression.TimeSeriesRegressionDataModule.plot parameter) TextCrossEntropyLoss (class in yamle.losses.classification) time_window (yamle.data.regression.TimeSeriesRegressionDataModule parameter) TimeSeriesRegressionDataModule (class in yamle.data.regression) TinyImageNet (class in yamle.third_party.tinyimagenet) TinyImageNetClassificationDataModule (class in yamle.data.classification) TinyImageNetRotationRegressionDataModule (class in yamle.data.regression) tokens_dim (yamle.models.mixer.MixerLayer parameter) (yamle.models.mixer.MixerModel parameter) tokens_hidden_dim (yamle.models.mixer.MixerLayer parameter) (yamle.models.mixer.MixerModel parameter) TorchtextClassificationDataModule (class in yamle.data.text) TorchtextClassificationModelIMDB (class in yamle.data.text) TorchtextClassificationModelWikiText103 (class in yamle.data.text) TorchtextClassificationModelWikiText2 (class in yamle.data.text) TorchvisionClassificationDataModule (class in yamle.data.classification) TorchvisionClassificationDataModuleCIFAR10 (class in yamle.data.classification) TorchvisionClassificationDataModuleCIFAR100 (class in yamle.data.classification) TorchvisionClassificationDataModuleCIFAR3 (class in yamle.data.classification) TorchvisionClassificationDataModuleCIFAR5 (class in yamle.data.classification) TorchvisionClassificationDataModuleFashionMNIST (class in yamle.data.classification) TorchvisionClassificationDataModuleMNIST (class in yamle.data.classification) TorchvisionClassificationDataModuleSVHN (class in yamle.data.classification) TorchvisionRegressionDataModule (class in yamle.data.regression) TorchvisionRotationRegressionDataModule (class in yamle.data.regression) TorchvisionRotationRegressionDataModuleCIFAR10 (class in yamle.data.regression) TorchvisionRotationRegressionDataModuleCIFAR100 (class in yamle.data.regression) TorchvisionRotationRegressionDataModuleFashionMNIST (class in yamle.data.regression) TorchvisionRotationRegressionDataModuleMNIST (class in yamle.data.regression) TorchvisionRotationRegressionDataModuleSVHN (class in yamle.data.regression) TorchvisionSegmentationDataModule (class in yamle.data.segmentation) TorchvisionSegmentationDataModuleCityscapes (class in yamle.data.segmentation) total_dataset_size() (yamle.data.datamodule.BaseDataModule method) ToyRegressionDataModule (class in yamle.data.regression) ToyTwoCirclesClassificationDataModule (class in yamle.data.classification) ToyTwoMoonsClassificationDataModule (class in yamle.data.classification) trace_input_output_shapes() (in module yamle.utils.tracing_utils) train (yamle.third_party.nyuv2.NYUv2 parameter) train_dataloader (yamle.trainers.calibration.CalibrationTrainer.fit parameter) train_dataloader() (yamle.data.datamodule.BaseDataModule method) train_dataset() (yamle.data.datamodule.BaseDataModule method) train_dataset_size() (yamle.data.datamodule.BaseDataModule method) train_joint_transform (yamle.data.datamodule.BaseDataModule parameter) train_joint_transform() (yamle.data.datamodule.BaseDataModule method) train_number_of_batches() (yamle.data.datamodule.BaseDataModule method) train_size (yamle.data.datamodule.BaseDataModule parameter) train_splits (yamle.data.datamodule.BaseDataModule parameter) train_splits_proportions (yamle.data.datamodule.BaseDataModule parameter) train_target_transform (yamle.data.datamodule.BaseDataModule parameter) train_target_transform() (yamle.data.datamodule.BaseDataModule method) train_tranform (yamle.data.datamodule.BaseDataModule parameter) (yamle.data.datamodule.VisionDataModule parameter) (yamle.data.segmentation.TorchvisionSegmentationDataModule parameter) train_transform() (yamle.data.datamodule.BaseDataModule method) trainer (yamle.utils.trainer_utils.GradientNormMonitorCallback.on_after_backward parameter) (yamle.utils.trainer_utils.GradientValueClippingCallback.on_after_backward parameter) (yamle.utils.trainer_utils.L1L2MonitorCallback.on_train_epoch_end parameter) (yamle.utils.trainer_utils.ValidationReporterMonitorCallback.on_validation_epoch_end parameter) trainer_factory() (in module yamle.trainers) trainer_type (yamle.trainers.trainer_factory parameter) training (yamle.losses.evidential_regression.NIG_NLL attribute) (yamle.losses.evidential_regression.NIG_Reg attribute) (yamle.methods.augmentation_classification.CutMixImageClassificationMethod attribute) (yamle.methods.augmentation_classification.CutOutImageClassificationMethod attribute) (yamle.methods.augmentation_classification.MixUpImageClassificationMethod attribute) (yamle.methods.augmentation_classification.RandomErasingImageClassificationMethod attribute) (yamle.methods.mimo.DataMUXMethod attribute) (yamle.methods.mimo.MIMMOMethod attribute) (yamle.methods.mimo.MIMOMethod attribute) (yamle.methods.mimo.MixMoMethod attribute) (yamle.methods.mimo.MixVitMethod attribute) (yamle.methods.mimo.UnMixMoMethod attribute) (yamle.methods.moe.MixtureOfExpertsMethod attribute) (yamle.methods.moe.MultiHeadEnsembleMethod attribute) (yamle.methods.pe.PEMethod attribute) (yamle.methods.rbnn.RBNNMethod attribute) (yamle.methods.sgld.SGLDMethod attribute) (yamle.methods.sngp.SNGPMethod attribute) (yamle.methods.svi.SVIFlipOutDropConnectMethod attribute) (yamle.methods.svi.SVIFlipOutRTMethod attribute) (yamle.methods.svi.SVILRTMethod attribute) (yamle.methods.svi.SVILRTVDMethod attribute) (yamle.methods.svi.SVIReparameterizationMethod attribute) (yamle.methods.svi.SVIRTMethod attribute) (yamle.methods.swag.SWAGMethod attribute) (yamle.methods.temperature_scaling.TemperatureMethod attribute) (yamle.methods.uncertain_method.MCSamplingMethod attribute) (yamle.methods.uncertain_method.MemberMethod attribute) (yamle.methods.uncertain_method.SVIMethod attribute) (yamle.models.convnet.ConvNetModel attribute) (yamle.models.convnet.ResidualConvNetModel attribute) (yamle.models.densenet.DenseBlock attribute) (yamle.models.densenet.DenseLayer attribute) (yamle.models.densenet.DenseNetModel attribute) (yamle.models.densenet.EmptyBlock attribute) (yamle.models.densenet.Transition attribute) (yamle.models.fc.FCModel attribute) (yamle.models.fc.ResidualFCModel attribute) (yamle.models.gp.GPModel attribute) (yamle.models.mixer.MixerModel attribute) (yamle.models.model.BaseModel attribute) (yamle.models.operations.Activation attribute) (yamle.models.operations.Add attribute) (yamle.models.operations.CompletelySeparableConv2d attribute) (yamle.models.operations.Conv2dExtractor attribute) (yamle.models.operations.Conv2dNormActivation attribute) (yamle.models.operations.DepthwiseSeparableConv2d attribute) (yamle.models.operations.DoubleConv2d attribute) (yamle.models.operations.Lambda attribute) (yamle.models.operations.LinearExtractor attribute) (yamle.models.operations.LinearNormActivation attribute) (yamle.models.operations.LSTM attribute) (yamle.models.operations.MatrixMultiplication attribute) (yamle.models.operations.Multiply attribute) (yamle.models.operations.Normalization attribute) (yamle.models.operations.OutputActivation attribute) (yamle.models.operations.ParallelModel attribute) (yamle.models.operations.Pooling attribute) (yamle.models.operations.Reduction attribute) (yamle.models.operations.ReshapeInput attribute) (yamle.models.operations.ReshapeOutput attribute) (yamle.models.operations.ResidualLayer attribute) (yamle.models.operations.ScalarAdder attribute) (yamle.models.operations.ScalarMultiplier attribute) (yamle.models.operations.SqueezeAndExcitation attribute) (yamle.models.operations.Unsqueeze attribute) (yamle.models.resnet.BasicBlock attribute) (yamle.models.resnet.Blocks attribute) (yamle.models.resnet.Bottleneck attribute) (yamle.models.resnet.ResNetModel attribute) (yamle.models.rnn.InputRepeater attribute) (yamle.models.rnn.RNNAutoEncoderModel attribute) (yamle.models.rnn.RNNModel attribute) (yamle.models.transformer.Attention attribute) (yamle.models.transformer.PositionalEncoding attribute) (yamle.models.transformer.TransformerModel attribute) (yamle.models.unet.DownBlock attribute) (yamle.models.unet.UNetModel attribute) (yamle.models.unet.UpBlock attribute) (yamle.models.vgg.Blocks attribute) (yamle.models.vgg.EmptyBlock attribute) (yamle.models.vgg.VGGBlock attribute) (yamle.models.vgg.VGGModel attribute) (yamle.models.visual_transformer.SpatialPositionalEmbedding attribute) (yamle.models.visual_transformer.VisualTransformerModel attribute) (yamle.quantization.models.operations.QuantizableAdd attribute) (yamle.quantization.models.specific.mcdropout.QuantisedDropout1d attribute) (yamle.quantization.models.specific.mcdropout.QuantisedDropout2d attribute) (yamle.quantization.models.specific.mcdropout.QuantisedDropout3d attribute) training_num_members (yamle.methods.uncertain_method.MemberMethod attribute) training_step() (yamle.methods.method.BaseMethod method) TrainingModelSavingCallback (class in yamle.utils.trainer_utils) transform (yamle.data.dataset_wrappers.SurrogateDataset parameter) (yamle.data.regression.WikiFaceDataset parameter) TransformerEncoderLayer (class in yamle.models.transformer) TransformerModel (class in yamle.models.transformer) transforms (yamle.data.transforms.JointCompose parameter) Transition (class in yamle.models.densenet) tuning_results_file() (in module yamle.utils.file_utils)

U

UCIClassificationDataModule (class in yamle.data.classification) UCIRegressionDataModule (class in yamle.data.regression) UNetModel (class in yamle.models.unet) UnMixMoMethod (class in yamle.methods.mimo)

Unsqueeze (class in yamle.models.operations) UnstructuredMagnitudePruner (class in yamle.pruning.unstructured.magnitude) UpBlock (class in yamle.models.unet) url (yamle.third_party.tinyimagenet.TinyImageNet attribute)

V

validation_dataloader (yamle.trainers.calibration.CalibrationTrainer.fit parameter) validation_dataloader() (yamle.data.datamodule.BaseDataModule method) validation_dataset() (yamle.data.datamodule.BaseDataModule method) validation_dataset_size() (yamle.data.datamodule.BaseDataModule method) validation_joint_transform() (yamle.data.datamodule.BaseDataModule method) validation_number_of_batches() (yamle.data.datamodule.BaseDataModule method) validation_portion (yamle.data.datamodule.BaseDataModule parameter) (yamle.data.datamodule.VisionDataModule parameter) (yamle.data.text.TorchtextClassificationDataModule parameter) validation_step() (yamle.methods.method.BaseMethod method) validation_target_transform() (yamle.data.datamodule.BaseDataModule method)

validation_transform() (yamle.data.datamodule.BaseDataModule method) ValidationModelSavingCallback (class in yamle.utils.trainer_utils) ValidationReporterMonitorCallback (class in yamle.utils.trainer_utils) value (yamle.data.augmentations.RandomErasing parameter) values (yamle.utils.plotting_utils.pareto_optimal parameter) VGGBlock (class in yamle.models.vgg) VGGModel (class in yamle.models.vgg) VisionClassificationDataModule (class in yamle.data.datamodule) VisionCorruption (class in yamle.third_party.imagenet_c) VisionDataModule (class in yamle.data.datamodule) VisionRegressionDataModule (class in yamle.data.datamodule) VisualTransformerModel (class in yamle.models.visual_transformer)

W

warm_starting_epochs (yamle.methods.dun.DUNMethod parameter) (yamle.methods.mimo.MIMMOMethod parameter) weight_bits (yamle.quantization.quantizer.BaseQuantizer parameter) WeightDecayRegularizer (class in yamle.regularizers.weight) weighted_regression_variance() (in module yamle.utils.operation_utils) WeightHistogramMonitorCallback (class in yamle.utils.trainer_utils) weights (yamle.utils.operation_utils.average_predictions parameter) (yamle.utils.operation_utils.classification_uncertainty_decomposition parameter) (yamle.utils.operation_utils.regression_uncertainty_decomposition parameter) (yamle.utils.operation_utils.weighted_regression_variance parameter) width (yamle.data.transforms.JointCenterCrop parameter) (yamle.data.transforms.JointResize parameter)

width_multiplier (yamle.models.densenet.DenseNetModel parameter) (yamle.models.fc.FCModel parameter) (yamle.models.resnet.ResNetModel parameter) (yamle.models.rnn.RNNAutoEncoderModel parameter) (yamle.models.rnn.RNNModel parameter) (yamle.models.vgg.VGGModel parameter) (yamle.models.visual_transformer.VisualTransformerModel parameter) WikiFaceDataset (class in yamle.data.regression) WikiFaceRegressionDataModule (class in yamle.data.regression) WineQualityUCIRegressionDataModule (class in yamle.data.regression)

X

x (yamle.models.convnet.ConvNetModel.forward parameter) (yamle.models.densenet.DenseNetModel.forward parameter) (yamle.models.fc.FCModel.forward parameter) (yamle.models.gp.GPModel.forward parameter) (yamle.models.mixer.MixerModel.forward parameter) (yamle.models.operations.LSTM.forward parameter) (yamle.models.operations.output_activation parameter) (yamle.models.operations.ScalarAdder.forward parameter) (yamle.models.operations.ScalarMultiplier.forward parameter) (yamle.models.resnet.ResNetModel.forward parameter) (yamle.models.rnn.InputRepeater.forward parameter) (yamle.models.rnn.RNNAutoEncoderModel.forward parameter) (yamle.models.rnn.RNNModel.forward parameter) (yamle.models.transformer.TransformerModel.forward parameter) (yamle.models.unet.UNetModel.forward parameter) (yamle.models.vgg.VGGModel.forward parameter) (yamle.models.visual_transformer.VisualTransformerModel.forward parameter) (yamle.third_party.imagenet_c.tabular_corruptions.additive_gaussian_noise parameter)

Y

YachtUCIRegressionDataModule (class in yamle.data.regression) yamle module yamle.data module yamle.data.augmentations module yamle.data.classification module yamle.data.custom module yamle.data.datamodule module yamle.data.dataset_wrappers module yamle.data.depth module yamle.data.reconstruction module yamle.data.regression module yamle.data.segmentation module yamle.data.text module yamle.data.transforms module yamle.defaults module yamle.losses module yamle.losses.classification module yamle.losses.contrastive module yamle.losses.evidential_regression module yamle.losses.loss module yamle.losses.regression module yamle.losses.segmentation module yamle.methods module yamle.methods.augmentation_classification module yamle.methods.be module yamle.methods.contrastive module yamle.methods.delta_uq module yamle.methods.dun module yamle.methods.early_exit module yamle.methods.ensemble module yamle.methods.evidential_regression module yamle.methods.gp module yamle.methods.laplace module yamle.methods.mcdropout module yamle.methods.method module yamle.methods.mimo module yamle.methods.moe module yamle.methods.pe module yamle.methods.rbnn module yamle.methods.sgld module yamle.methods.sngp module yamle.methods.svi module yamle.methods.swag module yamle.methods.temperature_scaling module yamle.methods.uncertain_method module yamle.models module yamle.models.convnet module yamle.models.densenet module yamle.models.fc module yamle.models.gp module yamle.models.mixer module yamle.models.model module yamle.models.operations module

yamle.models.resnet module yamle.models.rnn module yamle.models.transformer module yamle.models.unet module yamle.models.vgg module yamle.models.visual_transformer module yamle.pruning module yamle.pruning.pruner module yamle.pruning.unstructured module yamle.pruning.unstructured.magnitude module yamle.quantization module yamle.quantization.models module yamle.quantization.models.operations module yamle.quantization.models.specific module yamle.quantization.models.specific.mcdropout module yamle.quantization.qat module yamle.quantization.quantizer module yamle.quantization.static module yamle.regularizers module yamle.regularizers.feature module yamle.regularizers.gradient module yamle.regularizers.model module yamle.regularizers.regularizer module yamle.regularizers.weight module yamle.third_party module yamle.third_party.imagenet_c module yamle.third_party.imagenet_c.corruptions module yamle.third_party.imagenet_c.extra module yamle.third_party.imagenet_c.tabular_corruptions module yamle.third_party.medmnist module yamle.third_party.nyuv2 module yamle.third_party.tinyimagenet module yamle.trainers module yamle.trainers.calibration module yamle.trainers.ensemble module yamle.trainers.trainer module yamle.utils module yamle.utils.cli_utils module yamle.utils.export_utils module yamle.utils.file_utils module yamle.utils.operation_utils module yamle.utils.optimization_utils module yamle.utils.plotting_utils module yamle.utils.pruning_utils module yamle.utils.regularizer_utils module yamle.utils.running_utils module yamle.utils.specific module yamle.utils.specific.mimo_experiments module yamle.utils.specific.mimo_experiments.plotting_utils module yamle.utils.tracing_utils module yamle.utils.trainer_utils module yamle.utils.tuning_utils module

Z

zip_md5 (yamle.third_party.tinyimagenet.TinyImageNet attribute)

zoom_blur() (in module yamle.third_party.imagenet_c.corruptions)