All modules for which code is available
- autrainer.augmentations.abstract_augmentation
- autrainer.augmentations.augmentation_manager
- autrainer.augmentations.augmentation_pipeline
- autrainer.augmentations.augmentation_wrappers
- autrainer.augmentations.choice_augmentation
- autrainer.augmentations.image_augmentations
- autrainer.augmentations.sequential_augmentation
- autrainer.augmentations.spectrogram_augmentations
- autrainer.core.constants.abstract_constants
- autrainer.core.constants.export_constants
- autrainer.core.constants.naming_constants
- autrainer.core.constants.training_constants
- autrainer.core.plotting.plot_base
- autrainer.core.plotting.plot_metrics
- autrainer.core.scripts.create_script
- autrainer.core.scripts.delete_failed_script
- autrainer.core.scripts.delete_states_script
- autrainer.core.scripts.fetch_script
- autrainer.core.scripts.group_script
- autrainer.core.scripts.inference_script
- autrainer.core.scripts.postprocess_script
- autrainer.core.scripts.preprocess_script
- autrainer.core.scripts.show_script
- autrainer.core.scripts.train_script
- autrainer.core.utils.bookkeeping
- autrainer.core.utils.hardware
- autrainer.core.utils.set_seed
- autrainer.core.utils.silence
- autrainer.core.utils.timer
- autrainer.core.wrappers.instantiate
- autrainer.core.wrappers.main
- autrainer.criterions.criterion_wrappers
- autrainer.datasets.abstract_dataset
- autrainer.datasets.aibo
- autrainer.datasets.dcase_2016_t1
- autrainer.datasets.dcase_2018_t3
- autrainer.datasets.dcase_2020_t1a
- autrainer.datasets.edansa2019
- autrainer.datasets.emo_db
- autrainer.datasets.speech_commands
- autrainer.datasets.toy_dataset
- autrainer.datasets.utils.dataset_wrapper
- autrainer.datasets.utils.file_handlers
- autrainer.datasets.utils.target_transforms.abstract_target_transform
- autrainer.datasets.utils.target_transforms.label_encoder
- autrainer.datasets.utils.target_transforms.min_max_scaler
- autrainer.datasets.utils.target_transforms.multi_label_encoder
- autrainer.datasets.utils.zip_download_manager
- autrainer.loggers.abstract_logger
- autrainer.loggers.fallback_logger
- autrainer.loggers.mlflow_logger
- autrainer.loggers.tensorboard_logger
- autrainer.metrics.abstract_metric
- autrainer.metrics.classification
- autrainer.metrics.multi_label_classification
- autrainer.metrics.regression
- autrainer.models.abstract_model
- autrainer.models.ast_model
- autrainer.models.cnn_10
- autrainer.models.cnn_14
- autrainer.models.end2you
- autrainer.models.ffnn
- autrainer.models.leaf
- autrainer.models.sequential
- autrainer.models.tdnn
- autrainer.models.timm_wrapper
- autrainer.models.torchvision_wrapper
- autrainer.models.w2v2
- autrainer.models.whisper
- autrainer.optimizers.sam
- autrainer.postprocessing.aggregate
- autrainer.postprocessing.group
- autrainer.postprocessing.summarize
- autrainer.serving.serving
- autrainer.training.callback_manager
- autrainer.training.training
- autrainer.transforms.abstract_transform
- autrainer.transforms.smart_compose
- autrainer.transforms.specific_transforms
- autrainer.transforms.transform_manager