yamle.utils package

Subpackages

• yamle.utils.specific package
  • Subpackages
    • yamle.utils.specific.mimo_experiments package
      • Submodules

Submodules

• yamle.utils.cli_utils module
  • add_shared_args()
  • add_train_args()
  • add_test_args()
• yamle.utils.export_utils module
  • export_onnx()
• 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
  • average_predictions()
  • weighted_regression_variance()
  • repeat_inputs()
  • repeat_inputs_in_batch()
  • entropy()
  • classification_uncertainty_decomposition()
  • regression_uncertainty_decomposition()
• yamle.utils.optimization_utils module
  • get_optimizer()
  • get_scheduler()
  • ScalarScheduler
    • ScalarScheduler.step()
    • ScalarScheduler.get_value()
    • ScalarScheduler.reset()
    • ScalarScheduler.state_dict()
    • ScalarScheduler.load_state_dict()
  • LinearScalarScheduler
    • LinearScalarScheduler.step()
    • LinearScalarScheduler.set_hard_value()
    • LinearScalarScheduler.get_value()
    • LinearScalarScheduler.reset()
    • LinearScalarScheduler.state_dict()
    • LinearScalarScheduler.load_state_dict()
  • PowerGrowthScalarScheduler
    • PowerGrowthScalarScheduler.step()
    • PowerGrowthScalarScheduler.set_hard_value()
    • PowerGrowthScalarScheduler.get_value()
    • PowerGrowthScalarScheduler.reset()
    • PowerGrowthScalarScheduler.state_dict()
    • PowerGrowthScalarScheduler.load_state_dict()
  • SineScalarScheduler
    • SineScalarScheduler.step()
    • SineScalarScheduler.set_hard_value()
    • SineScalarScheduler.get_value()
    • SineScalarScheduler.reset()
    • SineScalarScheduler.state_dict()
    • SineScalarScheduler.load_state_dict()
  • CosineScalarScheduler
    • CosineScalarScheduler.step()
    • CosineScalarScheduler.set_hard_value()
    • CosineScalarScheduler.get_value()
    • CosineScalarScheduler.reset()
    • CosineScalarScheduler.state_dict()
    • CosineScalarScheduler.load_state_dict()
  • recover_frozen_weights()
  • freeze_weights()
  • split_optimizer_parameters()
  • set_optimizer_id()
  • disable_optimization()
  • enable_optimization()
• yamle.utils.plotting_utils module
  • average_augmentations()
  • fetch_value_and_std_from_result()
  • pareto_optimal()
• yamle.utils.pruning_utils module
  • enable_pruning()
  • disable_pruning()
  • is_layer_prunable()
  • is_parameter_prunable()
  • get_all_prunable_weights()
  • get_all_prunable_modules()
• yamle.utils.regularizer_utils module
  • disable_regularizer()
  • enable_regularizer()
• yamle.utils.running_utils module
  • extract_kwargs()
  • prepare_datamodule_kwargs()
  • prepare_model_kwargs()
  • prepare_loss_kwargs()
  • prepare_regularizer_kwargs()
  • prepare_metrics_kwargs()
  • prepare_method_kwargs()
  • prepare_trainer_kwargs()
  • prepare_test_trainer_kwargs()
  • prepare_pruner_kwargs()
  • prepare_quantizer_kwargs()
• yamle.utils.tracing_utils module
  • forward_shape_hook()
  • get_sample_input_and_target()
  • get_input_shape_from_model()
  • get_output_shape_from_model()
  • trace_input_output_shapes()
  • name_all_modules()
• yamle.utils.trainer_utils module
  • ValidationReporterMonitorCallback
    • ValidationReporterMonitorCallback.on_validation_epoch_end()
  • GradientNormMonitorCallback
    • GradientNormMonitorCallback.on_after_backward()
  • GradientValueClippingCallback
    • GradientValueClippingCallback.on_after_backward()
  • L1L2MonitorCallback
    • L1L2MonitorCallback.on_train_epoch_end()
  • WeightHistogramMonitorCallback
    • WeightHistogramMonitorCallback.on_train_epoch_end()
  • RegularizedWeightsMonitorCallback
    • RegularizedWeightsMonitorCallback.on_fit_start()
  • SplitParametersMonitorCallback
    • SplitParametersMonitorCallback.on_fit_start()
  • NoOptimizationMonitorCallback
    • NoOptimizationMonitorCallback.on_fit_start()
  • LoggingCallback
    • LoggingCallback.on_train_batch_end()
    • LoggingCallback.on_validation_batch_end()
    • LoggingCallback.on_test_batch_end()
    • LoggingCallback.on_train_epoch_start()
    • LoggingCallback.on_train_epoch_end()
    • LoggingCallback.on_validation_epoch_start()
    • LoggingCallback.on_validation_epoch_end()
    • LoggingCallback.on_test_epoch_start()
    • LoggingCallback.on_test_epoch_end()
  • ValidationModelSavingCallback
    • ValidationModelSavingCallback.on_validation_epoch_end()
  • TrainingModelSavingCallback
    • TrainingModelSavingCallback.on_train_epoch_end()
  • InitialModelSavingCallback
    • InitialModelSavingCallback.on_fit_start()
• yamle.utils.tuning_utils module
  • sample_initial_random_configs()
  • best_config_to_command_arguments()
  • plot_different_runs_and_metrics()
  • plot_different_runs_and_metric_config_combinations()
  • plot_different_metrics_and_trial_id()