Index

_

__init__() (mixmo.augmentations.augmix.AugMix method) (mixmo.augmentations.standard_augmentations.CustomCompose method) (mixmo.core.loss.AbstractLoss method) (mixmo.core.loss.SoftCrossEntropyLoss method) (mixmo.core.loss.WrapperLoss method) (mixmo.core.metrics_ensemble.MetricsEnsemble method) (mixmo.core.metrics_wrapper.MetricsWrapper method) (mixmo.core.scheduler.GradualWarmupScheduler method) (mixmo.core.scheduler.MultiGammaStepLR method) (mixmo.core.temperature_scaling.NetworkWithTemperature method) (mixmo.learners.abstract_learner.AbstractLearner method) (mixmo.learners.learner.Learner method) (mixmo.learners.model_wrapper.ModelWrapper method) (mixmo.loaders.abstract_loader.AbstractDataLoader method) (mixmo.loaders.batch_repetition_sampler.BatchRepetitionSampler method) (mixmo.loaders.cifar_dataset.CIFARCorruptions method) (mixmo.loaders.cifar_dataset.CustomCIFAR10 method) (mixmo.loaders.cifar_dataset.CustomCIFAR100 method) (mixmo.loaders.dataset_wrapper.DADataset method) (mixmo.loaders.dataset_wrapper.MixMoDataset method) (mixmo.loaders.dataset_wrapper.MSDADataset method) (mixmo.loaders.loader.CIFAR100Loader method) (mixmo.loaders.loader.CIFAR10Loader method) (mixmo.loaders.loader.TinyImagenet200Loader method) (mixmo.networks.resnet.PreActBlock method) (mixmo.networks.resnet.PreActResNet method) (mixmo.networks.resnet.PreActResNetMixMo method) (mixmo.networks.wrn.WideBasic method) (mixmo.networks.wrn.WideResNet method) (mixmo.networks.wrn.WideResNetMixMo method) _calculate_correct_fan() (in module mixmo.utils.torchutils) _calculate_fan_in_and_fan_out() (in module mixmo.utils.torchutils) _channel_mask() (in module mixmo.augmentations.mixing_blocks) _clean_metrics() (in module mixmo.core.metrics_wrapper) _compute_diversity() (mixmo.learners.model_wrapper.ModelWrapper method) _cow_mask() (in module mixmo.augmentations.mixing_blocks) _create_model_wrapper() (mixmo.learners.abstract_learner.AbstractLearner method) _cutmix_mask() (in module mixmo.augmentations.mixing_blocks) _define_config() (mixmo.networks.resnet.PreActResNet method) _forward() (mixmo.core.loss.SoftCrossEntropyLoss method) (mixmo.core.loss.WrapperLoss method)

_forward_subloss() (mixmo.core.loss.WrapperLoss method) _gaussian_blur_kernel() (in module mixmo.augmentations.mixing_blocks) _get_diversity_stats() (mixmo.core.metrics_wrapper.MetricsWrapper method) _init_block() (mixmo.networks.resnet.PreActResNet method) (mixmo.networks.wrn.WideResNet method) _init_core_network() (mixmo.networks.resnet.PreActResNet method) (mixmo.networks.wrn.WideResNet method) _init_dict_output_mixmo() (mixmo.loaders.dataset_wrapper.MixMoDataset method) _init_final_classifier() (mixmo.networks.resnet.PreActResNet method) (mixmo.networks.resnet.PreActResNetMixMo method) (mixmo.networks.wrn.WideResNetMixMo method) _init_first_layer() (mixmo.networks.resnet.PreActResNetMixMo method) _init_get_losses() (mixmo.core.loss.WrapperLoss method) _init_train_loader() (mixmo.loaders.abstract_loader.AbstractDataLoader method) _init_valtest_loader() (mixmo.loaders.abstract_loader.AbstractDataLoader method) _init_weights_resnet() (mixmo.networks.resnet.PreActResNet method) _learn_temperature() (mixmo.core.temperature_scaling.NetworkWithTemperature method) _make_layer() (mixmo.networks.resnet.PreActResNet method) (mixmo.networks.wrn.WideResNet method) _mixup_mask() (in module mixmo.augmentations.mixing_blocks) _n_cutinmix_mask() (in module mixmo.augmentations.mixing_blocks) _n_mix() (in module mixmo.augmentations.mixing_blocks) _n_mixup_mask() (in module mixmo.augmentations.mixing_blocks) _noise_mask() (in module mixmo.augmentations.mixing_blocks) _patchup_mask() (in module mixmo.augmentations.mixing_blocks) _patchuphard2d_mask() (in module mixmo.augmentations.mixing_blocks) _patchupsoft_mask() (in module mixmo.augmentations.mixing_blocks) _prepare_batch_test() (mixmo.learners.learner.Learner method) _prepare_batch_train() (mixmo.learners.learner.Learner method) _print_dict() (in module mixmo.utils.misc) _process_predictions() (in module mixmo.core.metrics_ensemble) _rand_bbox_of_area_lam() (in module mixmo.augmentations.mixing_blocks) _single_mix() (in module mixmo.augmentations.mixing_blocks) _stack0_mask() (in module mixmo.augmentations.mixing_blocks) _stack2_mask() (in module mixmo.augmentations.mixing_blocks) _stack_mask() (in module mixmo.augmentations.mixing_blocks) _subloop() (mixmo.learners.learner.Learner method) _target_balancing() (mixmo.loaders.dataset_wrapper.MixMoDataset method) _train_subloop() (mixmo.learners.learner.Learner method)

A

AbstractDataLoader (class in mixmo.loaders.abstract_loader) AbstractLearner (class in mixmo.learners.abstract_learner) AbstractLoss (class in mixmo.core.loss) apply_temperature() (mixmo.core.temperature_scaling.NetworkWithTemperature static method)

apply_temperature_on_logits() (in module mixmo.core.temperature_scaling) augment_list() (in module mixmo.augmentations.augmix) AugMix (class in mixmo.augmentations.augmix) AutoContrast() (in module mixmo.augmentations.augmix)

B

BatchRepetitionSampler (class in mixmo.loaders.batch_repetition_sampler)

Brightness() (in module mixmo.augmentations.augmix)

C

calibrate_via_tempscale() (mixmo.learners.model_wrapper.ModelWrapper method) call_dataset() (mixmo.loaders.dataset_wrapper.DADataset method) call_msda() (mixmo.loaders.dataset_wrapper.MSDADataset method) CIFAR100Loader (class in mixmo.loaders.loader) CIFAR10Loader (class in mixmo.loaders.loader) CIFARCorruptions (class in mixmo.loaders.cifar_dataset) clean_startswith() (in module mixmo.utils.misc) clean_update() (in module mixmo.utils.misc)

Color() (in module mixmo.augmentations.augmix) compute_mean_without_diagonal() (in module mixmo.core.metrics_ensemble) compute_pairwise_diversity() (in module mixmo.core.metrics_ensemble) Contrast() (in module mixmo.augmentations.augmix) csv_writter() (in module mixmo.utils.misc) CustomCIFAR10 (class in mixmo.loaders.cifar_dataset) CustomCIFAR100 (class in mixmo.loaders.cifar_dataset) CustomCompose (class in mixmo.augmentations.standard_augmentations)

D

DADataset (class in mixmo.loaders.dataset_wrapper)

dirty_update() (in module mixmo.utils.misc)

E

Equalize() (in module mixmo.augmentations.augmix)

evaluate() (mixmo.learners.abstract_learner.AbstractLearner method) evaluate_loop() (mixmo.learners.learner.Learner method)

F

find_best_epoch() (in module mixmo.utils.misc) forward() (mixmo.core.loss.AbstractLoss method) (mixmo.core.temperature_scaling.NetworkWithTemperature method) (mixmo.networks.resnet.PreActBlock method) (mixmo.networks.resnet.PreActResNet method) (mixmo.networks.wrn.WideBasic method)

G

get_accumulator_stats() (mixmo.core.loss.AbstractLoss method) (mixmo.core.loss.WrapperLoss method) get_accuracy_multi() (in module mixmo.core.metrics_ensemble) get_brier() (in module mixmo.core.metrics_wrapper) get_checkpoint() (in module mixmo.utils.misc) get_default_composed_augmentations() (in module mixmo.augmentations.standard_augmentations) get_dict_to_scores() (mixmo.learners.model_wrapper.ModelWrapper method) get_ece() (in module mixmo.core.metrics_wrapper) get_individualaccuracies() (mixmo.core.metrics_ensemble.MetricsEnsemble method) get_ll() (in module mixmo.core.metrics_wrapper) get_loader() (in module mixmo.loaders) get_logger() (in module mixmo.utils.logger) get_logs_path() (in module mixmo.utils.misc) get_loss() (in module mixmo.core.loss) get_mixmo_mix_method_at_ratio_epoch() (mixmo.loaders.dataset_wrapper.MixMoDataset method)

get_model_path() (in module mixmo.utils.misc) get_network() (in module mixmo.networks) get_nprandom() (in module mixmo.utils.misc) get_optimizer() (in module mixmo.core.optimizer) get_output_folder_from_config() (in module mixmo.utils.misc) get_predictions() (in module mixmo.learners.model_wrapper) get_previous_ckpt() (in module mixmo.utils.misc) get_random() (in module mixmo.utils.misc) get_scheduler() (in module mixmo.core.scheduler) get_scores() (mixmo.core.metrics_wrapper.MetricsWrapper method) get_short_logs() (mixmo.learners.model_wrapper.ModelWrapper method) get_tace_bayesgroup() (in module mixmo.core.metrics_wrapper) get_value_when_none() (in module mixmo.augmentations.augmix) get_warmup_scheduler() (in module mixmo.core.scheduler) GradualWarmupScheduler (class in mixmo.core.scheduler)

H

help() (mixmo.core.metrics_ensemble.MetricsEnsemble method)

I

Identity() (in module mixmo.augmentations.augmix) ifnotfound_update() (in module mixmo.utils.misc) Invert() (in module mixmo.augmentations.augmix) is_float() (in module mixmo.utils.misc)

is_int() (in module mixmo.utils.misc) is_nan() (in module mixmo.utils.misc) is_none() (in module mixmo.utils.misc) is_zero() (in module mixmo.utils.misc)

K

kaiming_normal_truncated() (in module mixmo.utils.torchutils)

L

l2_reg() (mixmo.core.loss.WrapperLoss method) Learner (class in mixmo.learners.learner)

load_checkpoint() (mixmo.learners.abstract_learner.AbstractLearner method) load_config_yaml() (in module mixmo.utils.misc) load_yaml() (in module mixmo.utils.misc)

M

make_corruptions_test_dataset() (mixmo.loaders.abstract_loader.AbstractDataLoader method) make_image() (in module mixmo.utils.visualize) make_standard_loader() (mixmo.loaders.abstract_loader.AbstractDataLoader method) merge_scores() (in module mixmo.core.metrics_wrapper) MetricsEnsemble (class in mixmo.core.metrics_ensemble) MetricsWrapper (class in mixmo.core.metrics_wrapper) mix() (in module mixmo.augmentations.mixing_blocks) mix_manifolds() (in module mixmo.augmentations.mixing_blocks) mixmo module, [1] mixmo.augmentations module, [1] mixmo.augmentations.augmix module, [1] mixmo.augmentations.mixing_blocks module, [1] mixmo.augmentations.standard_augmentations module, [1] mixmo.core module, [1] mixmo.core.loss module, [1] mixmo.core.metrics_ensemble module, [1] mixmo.core.metrics_wrapper module, [1] mixmo.core.optimizer module, [1] mixmo.core.scheduler module, [1] mixmo.core.temperature_scaling module, [1] mixmo.learners module mixmo.learners.abstract_learner module mixmo.learners.learner module mixmo.learners.model_wrapper module mixmo.loaders module, [1] mixmo.loaders.abstract_loader module, [1] mixmo.loaders.batch_repetition_sampler module, [1] mixmo.loaders.cifar_dataset module, [1] mixmo.loaders.dataset_wrapper module, [1] mixmo.loaders.loader module, [1] mixmo.networks module, [1]

mixmo.networks.resnet module, [1] mixmo.networks.wrn module, [1] mixmo.utils module, [1] mixmo.utils.config module, [1] mixmo.utils.logger module, [1] mixmo.utils.misc module, [1] mixmo.utils.torchsummary module, [1] mixmo.utils.torchutils module, [1] mixmo.utils.visualize module, [1] MixMoDataset (class in mixmo.loaders.dataset_wrapper) ModelWrapper (class in mixmo.learners.model_wrapper) module mixmo, [1] mixmo.augmentations, [1] mixmo.augmentations.augmix, [1] mixmo.augmentations.mixing_blocks, [1] mixmo.augmentations.standard_augmentations, [1] mixmo.core, [1] mixmo.core.loss, [1] mixmo.core.metrics_ensemble, [1] mixmo.core.metrics_wrapper, [1] mixmo.core.optimizer, [1] mixmo.core.scheduler, [1] mixmo.core.temperature_scaling, [1] mixmo.learners mixmo.learners.abstract_learner mixmo.learners.learner mixmo.learners.model_wrapper mixmo.loaders, [1] mixmo.loaders.abstract_loader, [1] mixmo.loaders.batch_repetition_sampler, [1] mixmo.loaders.cifar_dataset, [1] mixmo.loaders.dataset_wrapper, [1] mixmo.loaders.loader, [1] mixmo.networks, [1] mixmo.networks.resnet, [1] mixmo.networks.wrn, [1] mixmo.utils, [1] mixmo.utils.config, [1] mixmo.utils.logger, [1] mixmo.utils.misc, [1] mixmo.utils.torchsummary, [1] mixmo.utils.torchutils, [1] mixmo.utils.visualize, [1] MSDADataset (class in mixmo.loaders.dataset_wrapper) MultiGammaStepLR (class in mixmo.core.scheduler)

N

NetworkWithTemperature (class in mixmo.core.temperature_scaling)

O

onehot() (in module mixmo.utils.torchutils)

output_format() (mixmo.loaders.batch_repetition_sampler.BatchRepetitionSampler method)

P

Posterize() (in module mixmo.augmentations.augmix) PreActBlock (class in mixmo.networks.resnet) PreActResNet (class in mixmo.networks.resnet)

PreActResNetMixMo (class in mixmo.networks.resnet) predict() (mixmo.learners.model_wrapper.ModelWrapper method) print_args() (in module mixmo.utils.misc) print_dict() (in module mixmo.utils.misc)

R

random_lower_than() (in module mixmo.utils.misc) randperm_static() (in module mixmo.utils.torchutils)

ratio_errors() (in module mixmo.core.metrics_ensemble) Rotate() (in module mixmo.augmentations.augmix)

S

sample_lams() (in module mixmo.utils.misc) save_checkpoint() (mixmo.learners.abstract_learner.AbstractLearner method) save_tb() (mixmo.learners.abstract_learner.AbstractLearner method) scaled_network() (mixmo.learners.model_wrapper.ModelWrapper property) set_determ() (in module mixmo.utils.misc) Sharpness() (in module mixmo.augmentations.augmix) ShearX() (in module mixmo.augmentations.augmix)

ShearY() (in module mixmo.augmentations.augmix) show_batch() (in module mixmo.utils.visualize) show_metrics() (in module mixmo.core.metrics_wrapper) SoftCrossEntropyLoss (class in mixmo.core.loss) Solarize() (in module mixmo.augmentations.augmix) step() (mixmo.learners.model_wrapper.ModelWrapper method) summary() (in module mixmo.utils.torchsummary)

T

tb_logger() (mixmo.learners.abstract_learner.AbstractLearner property) TinyImagenet200Loader (class in mixmo.loaders.loader) to_eval_mode() (mixmo.learners.model_wrapper.ModelWrapper method) to_train_mode() (mixmo.learners.model_wrapper.ModelWrapper method)

train() (mixmo.learners.abstract_learner.AbstractLearner method) train_loop() (mixmo.learners.learner.Learner method) TranslateX() (in module mixmo.augmentations.augmix) TranslateY() (in module mixmo.augmentations.augmix) truncated_normal_() (in module mixmo.utils.torchutils)

U

update() (in module mixmo.utils.misc) (mixmo.core.metrics_wrapper.MetricsWrapper method)

update_diversity() (mixmo.core.metrics_wrapper.MetricsWrapper method)

W

weights_init_hetruncatednormal() (in module mixmo.utils.torchutils) WideBasic (class in mixmo.networks.wrn) WideResNet (class in mixmo.networks.wrn)

WideResNetMixMo (class in mixmo.networks.wrn) WrapperLoss (class in mixmo.core.loss) write_calibration() (in module mixmo.utils.visualize)