yamle package#
Subpackages#
- yamle.data package
data_factory()- Submodules
- yamle.data.augmentations module
- yamle.data.classification module
TorchvisionClassificationDataModuleTinyImageNetClassificationDataModuleTorchvisionClassificationDataModuleMNISTTorchvisionClassificationDataModuleFashionMNISTTorchvisionClassificationDataModuleSVHNTorchvisionClassificationDataModuleCIFAR10TorchvisionClassificationDataModuleCIFAR3TorchvisionClassificationDataModuleCIFAR5TorchvisionClassificationDataModuleCIFAR100MedMNISTClassificationDataModulePneumoniaMNISTClassificationDataModuleBreastMNISTClassificationDataModuleDermaMNISTClassificationDataModuleBloodMNISTClassificationDataModuleToyTwoMoonsClassificationDataModuleToyTwoCirclesClassificationDataModuleUCIClassificationDataModuleBreastCancerUCIClassificationDataModuleAdultIncomeUCIClassificationDataModuleCarEvaluationUCIClassificationDataModuleCreditUCIClassificationDataModuleDermatologyUCIClassificationDataModuleECG5000ClassificationDataModule
- 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
ToyRegressionDataModuleUCIRegressionDataModuleMedMNISTRegressionDataModuleRetinaMNISTDataModuleConcreteUCIRegressionDataModuleEnergyUCIRegressionDataModuleBostonUCIRegressionDataModuleWineQualityUCIRegressionDataModuleYachtUCIRegressionDataModuleAbaloneUCIRegressionDataModuleTelemonitoringUCIRegressionDataModuleTimeSeriesRegressionDataModuleTemperatureTimeSeriesDataModuleWikiFaceDatasetTorchvisionRegressionDataModuleWikiFaceRegressionDataModuleTorchvisionRotationRegressionDataModuleTinyImageNetRotationRegressionDataModuleTorchvisionRotationRegressionDataModuleMNISTTorchvisionRotationRegressionDataModuleFashionMNISTTorchvisionRotationRegressionDataModuleSVHNTorchvisionRotationRegressionDataModuleCIFAR10TorchvisionRotationRegressionDataModuleCIFAR100
- yamle.data.segmentation module
- yamle.data.text module
- yamle.data.transforms module
- yamle.losses package
- yamle.methods package
method_factory()- Submodules
- 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 package
model_factory()- Submodules
- 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
output_activation()OutputActivationReshapeOutputReshapeInputUnsqueezeAddMultiplyResidualLayerParallelModelNormalizationPoolingActivationMatrixMultiplicationLinearNormActivationConv2dNormActivationDoubleConv2dDepthwiseSeparableConv2dCompletelySeparableConv2dSqueezeAndExcitationLSTMLambdaReductionScalarMultiplierScalarAdderLinearExtractorConv2dExtractor
- 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 package
- yamle.quantization package
- yamle.regularizers package
- yamle.third_party package
- yamle.trainers package
- yamle.utils package
- Subpackages
- Submodules
- yamle.utils.cli_utils module
- yamle.utils.export_utils module
- yamle.utils.file_utils module
import_config_function_from_file()current_time()log_file()config_logger()create_experiment_folder()save_scripts()get_experiment_name()argparse_to_dictionary()argparse_to_command()argparse_to_config_file()parse_args()store_metrics()results_file()tuning_results_file()model_file()model_onnx_file()model_initial_file()model_quantized_file()model_best_on_validation_file()model_train_epoch_file()method_file()args_file()args_dictionary_file()plots_file()config_file()predictions_file()save_pickle()load_pickle()save_model()load_model()save_model_onnx()save_method()load_method()save_args()load_args()save_args_dictionary()load_args_dictionary()save_results()load_results()save_tuning_results()load_tuning_results()save_experiment()remove_argparse_argument()
- 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.tracing_utils module
- yamle.utils.trainer_utils module
ValidationReporterMonitorCallbackGradientNormMonitorCallbackGradientValueClippingCallbackL1L2MonitorCallbackWeightHistogramMonitorCallbackRegularizedWeightsMonitorCallbackSplitParametersMonitorCallbackNoOptimizationMonitorCallbackLoggingCallbackValidationModelSavingCallbackTrainingModelSavingCallbackInitialModelSavingCallback
- yamle.utils.tuning_utils module