Index

_

__call__() (IPSL_AID.loss.EDMLoss method) (IPSL_AID.loss.VELoss method) (IPSL_AID.loss.VPLoss method) __getitem__() (IPSL_AID.dataset.DataPreprocessor method) __init__() (IPSL_AID.dataset.DataPreprocessor method) (IPSL_AID.dataset.TestDataPreprocessor method) (IPSL_AID.diagnostics.TestPlottingFunctions method) (IPSL_AID.diagnostics.TestSSRFunction method) (IPSL_AID.evaluater.MetricTracker method) (IPSL_AID.evaluater.TestCRPSFunction method) (IPSL_AID.evaluater.TestDenormalizeFunction method) (IPSL_AID.evaluater.TestErrorMetrics method) (IPSL_AID.evaluater.TestMetricTracker method) (IPSL_AID.evaluater.TestRunValidation method) (IPSL_AID.logger.Logger method) (IPSL_AID.logger.TestLogger method) (IPSL_AID.loss.EDMLoss method) (IPSL_AID.loss.TestLosses method) (IPSL_AID.loss.UnetLoss method) (IPSL_AID.loss.VELoss method) (IPSL_AID.loss.VPLoss method) (IPSL_AID.model.TestModelLoader method) (IPSL_AID.model_utils.ModelUtils method) (IPSL_AID.model_utils.TestModel method) (IPSL_AID.model_utils.TestModelUtils method) (IPSL_AID.networks.Conv2d method) (IPSL_AID.networks.DhariwalUNet method) (IPSL_AID.networks.EDMPrecond method) (IPSL_AID.networks.FourierEmbedding method) (IPSL_AID.networks.GroupNorm method) (IPSL_AID.networks.iDDPMPrecond method) (IPSL_AID.networks.Linear method) (IPSL_AID.networks.PositionalEmbedding method) (IPSL_AID.networks.SongUNet method) (IPSL_AID.networks.TestDiffusionNetworks method) (IPSL_AID.networks.UNetBlock method) (IPSL_AID.networks.VEPrecond method) (IPSL_AID.networks.VPPrecond method) (IPSL_AID.utils.FileUtils method) (IPSL_AID.utils.TestEasyDict method) (IPSL_AID.utils.TestFileUtils method)

__len__() (IPSL_AID.dataset.DataPreprocessor method)

A

affine (IPSL_AID.networks.UNetBlock attribute)

alpha_bar() (IPSL_AID.networks.iDDPMPrecond method) AttentionOp (class in IPSL_AID.networks)

B

backward() (IPSL_AID.networks.AttentionOp static method) beta_d (IPSL_AID.loss.VPLoss attribute) beta_min (IPSL_AID.loss.VPLoss attribute) bias (IPSL_AID.networks.Conv2d attribute) (IPSL_AID.networks.GroupNorm attribute) (IPSL_AID.networks.Linear attribute)

BORDER_STYLE (IPSL_AID.diagnostics.PlotConfig attribute) BORDER_w (IPSL_AID.diagnostics.PlotConfig attribute)

C

calculate_pearsoncorr_nparray() (in module IPSL_AID.diagnostics) calculate_psd2d_simple() (in module IPSL_AID.diagnostics) center_tracker (IPSL_AID.dataset.DataPreprocessor attribute) clear_logs() (IPSL_AID.logger.Logger method) coarse_down_up() (in module IPSL_AID.dataset) COASTLINE_w (IPSL_AID.diagnostics.PlotConfig attribute) COLORBAR_h (IPSL_AID.diagnostics.PlotConfig attribute) COLORBAR_PAD (IPSL_AID.diagnostics.PlotConfig attribute) COLORMAPS (IPSL_AID.diagnostics.PlotConfig attribute)

const_vars (IPSL_AID.dataset.DataPreprocessor attribute) Conv2d (class in IPSL_AID.networks) convert_units() (IPSL_AID.diagnostics.PlotConfig class method) count (IPSL_AID.evaluater.MetricTracker attribute) count_parameters_by_layer() (IPSL_AID.model_utils.ModelUtils static method) create_data_loaders() (in module IPSL_AID.main) create_dummy_netcdf() (in module IPSL_AID.dataset) create_dummy_statistics_json() (in module IPSL_AID.dataset) crps_ensemble_all() (in module IPSL_AID.evaluater)

D

DataPreprocessor (class in IPSL_AID.dataset) day (IPSL_AID.dataset.DataPreprocessor attribute) dec (IPSL_AID.networks.DhariwalUNet attribute) (IPSL_AID.networks.SongUNet attribute) DEFAULT_FIGSIZE_MULTIPLIER (IPSL_AID.diagnostics.PlotConfig attribute)

DEFAULT_SAVE_DIR (IPSL_AID.diagnostics.PlotConfig attribute) denormalize() (in module IPSL_AID.evaluater) DhariwalUNet (class in IPSL_AID.networks) doy_norm (IPSL_AID.dataset.DataPreprocessor attribute) dry_frequency_map() (in module IPSL_AID.diagnostics)

E

EasyDict (class in IPSL_AID.utils) edm_sampler() (in module IPSL_AID.evaluater) EDMLoss (class in IPSL_AID.loss) EDMPrecond (class in IPSL_AID.networks) enc (IPSL_AID.networks.DhariwalUNet attribute) (IPSL_AID.networks.SongUNet attribute)

endpoint (IPSL_AID.networks.PositionalEmbedding attribute) epsilon_t (IPSL_AID.loss.VPLoss attribute) error() (IPSL_AID.logger.Logger method) eval_slices (IPSL_AID.dataset.DataPreprocessor attribute) exception() (IPSL_AID.logger.Logger method) extract_batch() (IPSL_AID.dataset.DataPreprocessor method), [1]

F

FileUtils (class in IPSL_AID.utils) filter_batch() (IPSL_AID.dataset.DataPreprocessor method), [1] FIXED_DIFF_RANGES (IPSL_AID.diagnostics.PlotConfig attribute) FIXED_DIFF_RANGES_ERRORS (IPSL_AID.diagnostics.PlotConfig attribute) FIXED_MAE_RANGES (IPSL_AID.diagnostics.PlotConfig attribute) FIXED_SSR_RANGES (IPSL_AID.diagnostics.PlotConfig attribute) forward() (IPSL_AID.model_utils.TestModel method) (IPSL_AID.networks.AttentionOp static method) (IPSL_AID.networks.Conv2d method) (IPSL_AID.networks.DhariwalUNet method) (IPSL_AID.networks.EDMPrecond method) (IPSL_AID.networks.FourierEmbedding method) (IPSL_AID.networks.GroupNorm method) (IPSL_AID.networks.iDDPMPrecond method) (IPSL_AID.networks.Linear method) (IPSL_AID.networks.PositionalEmbedding method) (IPSL_AID.networks.SongUNet method) (IPSL_AID.networks.UNetBlock method) (IPSL_AID.networks.VEPrecond method) (IPSL_AID.networks.VPPrecond method)

FourierEmbedding (class in IPSL_AID.networks) freqs (IPSL_AID.networks.FourierEmbedding attribute)

G

gaussian_filter() (in module IPSL_AID.dataset) generate_evaluation_slices() (IPSL_AID.dataset.DataPreprocessor method), [1] generate_random_batch_centers() (IPSL_AID.dataset.DataPreprocessor method), [1] generate_region_slices() (IPSL_AID.dataset.DataPreprocessor method) generate_residuals_norm() (in module IPSL_AID.evaluater) get_center_indices_from_latlon() (IPSL_AID.dataset.DataPreprocessor method) get_colormap() (IPSL_AID.diagnostics.PlotConfig class method) get_curl() (in module IPSL_AID.diagnostics) get_divergence() (in module IPSL_AID.diagnostics)

get_fixed_diff_range() (IPSL_AID.diagnostics.PlotConfig static method) get_fixed_diff_range_errors() (IPSL_AID.diagnostics.PlotConfig static method) get_fixed_mae_range() (IPSL_AID.diagnostics.PlotConfig static method) get_fixed_ssr_range() (IPSL_AID.diagnostics.PlotConfig static method) get_parameter_number() (IPSL_AID.model_utils.ModelUtils static method) get_plot_name() (IPSL_AID.diagnostics.PlotConfig class method) getmean() (IPSL_AID.evaluater.MetricTracker method) getsqrtmean() (IPSL_AID.evaluater.MetricTracker method) getstd() (IPSL_AID.evaluater.MetricTracker method) GroupNorm (class in IPSL_AID.networks)

H

hour (IPSL_AID.dataset.DataPreprocessor attribute)

hour_norm (IPSL_AID.dataset.DataPreprocessor attribute)

I

iDDPMPrecond (class in IPSL_AID.networks) img_height (IPSL_AID.networks.DhariwalUNet attribute) (IPSL_AID.networks.SongUNet attribute) img_resolution (IPSL_AID.networks.DhariwalUNet attribute) (IPSL_AID.networks.EDMPrecond attribute) (IPSL_AID.networks.iDDPMPrecond attribute) (IPSL_AID.networks.SongUNet attribute) (IPSL_AID.networks.VEPrecond attribute) (IPSL_AID.networks.VPPrecond attribute) img_width (IPSL_AID.networks.DhariwalUNet attribute) (IPSL_AID.networks.SongUNet attribute) in_channels (IPSL_AID.networks.EDMPrecond attribute) (IPSL_AID.networks.iDDPMPrecond attribute) (IPSL_AID.networks.VEPrecond attribute) (IPSL_AID.networks.VPPrecond attribute) info() (IPSL_AID.logger.Logger method) IPSL_AID.dataset module IPSL_AID.diagnostics module

IPSL_AID.download_ERA5_cds module IPSL_AID.evaluater module IPSL_AID.generate_all_data_ERA5 module IPSL_AID.logger module IPSL_AID.loss module IPSL_AID.main module IPSL_AID.model module IPSL_AID.model_utils module IPSL_AID.networks module IPSL_AID.utils module

K

kl_divergence_all() (in module IPSL_AID.evaluater)

L

label_dim (IPSL_AID.networks.EDMPrecond attribute) (IPSL_AID.networks.iDDPMPrecond attribute) (IPSL_AID.networks.VEPrecond attribute) (IPSL_AID.networks.VPPrecond attribute) label_dropout (IPSL_AID.networks.DhariwalUNet attribute) (IPSL_AID.networks.SongUNet attribute) LAKE_w (IPSL_AID.diagnostics.PlotConfig attribute) Linear (class in IPSL_AID.networks) load_checkpoint() (IPSL_AID.model_utils.ModelUtils static method)

load_dynamic_covariates() (IPSL_AID.dataset.DataPreprocessor method), [1] load_model_and_loss() (in module IPSL_AID.model) load_training_checkpoint() (IPSL_AID.model_utils.ModelUtils static method) log_configuration() (in module IPSL_AID.main) log_metrics() (IPSL_AID.logger.Logger method) log_model_summary() (IPSL_AID.model_utils.ModelUtils static method) Logger (class in IPSL_AID.logger) loss_fn (IPSL_AID.loss.UnetLoss attribute) loss_type (IPSL_AID.loss.UnetLoss attribute)

M

mae_all() (in module IPSL_AID.evaluater) main() (in module IPSL_AID.download_ERA5_cds) (in module IPSL_AID.generate_all_data_ERA5) (in module IPSL_AID.main) make_divisible_hw() (in module IPSL_AID.main) makedir() (IPSL_AID.utils.FileUtils static method) makefile() (IPSL_AID.utils.FileUtils static method) map_augment (IPSL_AID.networks.DhariwalUNet attribute) (IPSL_AID.networks.SongUNet attribute) map_label (IPSL_AID.networks.DhariwalUNet attribute) (IPSL_AID.networks.SongUNet attribute) map_noise (IPSL_AID.networks.DhariwalUNet attribute) (IPSL_AID.networks.SongUNet attribute) max_positions (IPSL_AID.networks.PositionalEmbedding attribute) MetricTracker (class in IPSL_AID.evaluater) model (IPSL_AID.networks.EDMPrecond attribute) (IPSL_AID.networks.iDDPMPrecond attribute) (IPSL_AID.networks.VEPrecond attribute) (IPSL_AID.networks.VPPrecond attribute)

ModelUtils (class in IPSL_AID.model_utils) module IPSL_AID.dataset IPSL_AID.diagnostics IPSL_AID.download_ERA5_cds IPSL_AID.evaluater IPSL_AID.generate_all_data_ERA5 IPSL_AID.logger IPSL_AID.loss IPSL_AID.main IPSL_AID.model IPSL_AID.model_utils IPSL_AID.networks IPSL_AID.utils month (IPSL_AID.dataset.DataPreprocessor attribute)

N

new_epoch() (IPSL_AID.dataset.DataPreprocessor method), [1] nmae_all() (in module IPSL_AID.evaluater)

normalize() (IPSL_AID.dataset.DataPreprocessor method), [1] normalize_time() (IPSL_AID.dataset.DataPreprocessor method), [1] num_channels (IPSL_AID.networks.PositionalEmbedding attribute)

O

out_channels (IPSL_AID.networks.EDMPrecond attribute) (IPSL_AID.networks.iDDPMPrecond attribute) (IPSL_AID.networks.VEPrecond attribute) (IPSL_AID.networks.VPPrecond attribute)

out_conv (IPSL_AID.networks.DhariwalUNet attribute) out_norm (IPSL_AID.networks.DhariwalUNet attribute)

P

P_mean (IPSL_AID.loss.EDMLoss attribute) P_std (IPSL_AID.loss.EDMLoss attribute) parse_args() (in module IPSL_AID.download_ERA5_cds) (in module IPSL_AID.generate_all_data_ERA5) (in module IPSL_AID.main) pearson_all() (in module IPSL_AID.evaluater) plot_average_metrics() (in module IPSL_AID.diagnostics) plot_comparison_hexbin() (in module IPSL_AID.diagnostics) plot_dry_frequency_map() (in module IPSL_AID.diagnostics) plot_ensemble_surface() (in module IPSL_AID.diagnostics) plot_error_map() (in module IPSL_AID.diagnostics) plot_global_surface_robinson() (in module IPSL_AID.diagnostics) plot_loss_histories() (in module IPSL_AID.diagnostics) plot_MAE_map() (in module IPSL_AID.diagnostics) plot_mean_curl_map() (in module IPSL_AID.diagnostics) plot_mean_divergence_map() (in module IPSL_AID.diagnostics) plot_metric_histories() (in module IPSL_AID.diagnostics)

plot_metrics_heatmap() (in module IPSL_AID.diagnostics) plot_power_spectra() (in module IPSL_AID.diagnostics) plot_qq_quantiles() (in module IPSL_AID.diagnostics) plot_ranks() (in module IPSL_AID.diagnostics) plot_spatiotemporal_histograms() (in module IPSL_AID.diagnostics) plot_spread_skill_ratio_hexbin() (in module IPSL_AID.diagnostics) plot_spread_skill_ratio_map() (in module IPSL_AID.diagnostics) plot_surface() (in module IPSL_AID.diagnostics) plot_temporal_series_comparison() (in module IPSL_AID.diagnostics) plot_validation_hexbin() (in module IPSL_AID.diagnostics) plot_validation_mvcorr() (in module IPSL_AID.diagnostics) plot_validation_mvcorr_space() (in module IPSL_AID.diagnostics) plot_validation_pdfs() (in module IPSL_AID.diagnostics) plot_zoom_comparison() (in module IPSL_AID.diagnostics) PlotConfig (class in IPSL_AID.diagnostics) PositionalEmbedding (class in IPSL_AID.networks) print_model_layers() (IPSL_AID.model_utils.ModelUtils static method)

R

r2_all() (in module IPSL_AID.evaluater) radial_average_psd() (in module IPSL_AID.diagnostics) random_centers (IPSL_AID.dataset.DataPreprocessor attribute) ranks() (in module IPSL_AID.diagnostics) reconstruct_original_layout() (in module IPSL_AID.evaluater) reduction (IPSL_AID.loss.UnetLoss attribute) resample_filter (IPSL_AID.networks.Conv2d attribute)

reset() (IPSL_AID.evaluater.MetricTracker method) resolve_region_center() (in module IPSL_AID.main) rmse_all() (in module IPSL_AID.evaluater) round_sigma() (IPSL_AID.networks.EDMPrecond method) (IPSL_AID.networks.iDDPMPrecond method) (IPSL_AID.networks.VEPrecond method) (IPSL_AID.networks.VPPrecond method) run_validation() (in module IPSL_AID.evaluater)

S

sample_random_time_indices() (IPSL_AID.dataset.DataPreprocessor method), [1] sample_time_steps_by_doy() (IPSL_AID.dataset.DataPreprocessor method), [1] sampler() (in module IPSL_AID.evaluater) save_checkpoint() (IPSL_AID.model_utils.ModelUtils static method) save_emergency_checkpoint() (IPSL_AID.model_utils.ModelUtils static method) save_training_checkpoint() (IPSL_AID.model_utils.ModelUtils static method) setUp() (IPSL_AID.dataset.TestDataPreprocessor method) (IPSL_AID.diagnostics.TestPlottingFunctions method) (IPSL_AID.diagnostics.TestSSRFunction method) (IPSL_AID.evaluater.TestCRPSFunction method) (IPSL_AID.evaluater.TestDenormalizeFunction method) (IPSL_AID.evaluater.TestErrorMetrics method) (IPSL_AID.evaluater.TestMetricTracker method) (IPSL_AID.evaluater.TestRunValidation method) (IPSL_AID.logger.TestLogger method) (IPSL_AID.loss.TestLosses method) (IPSL_AID.model.TestModelLoader method) (IPSL_AID.model_utils.TestModelUtils method) (IPSL_AID.networks.TestDiffusionNetworks method) (IPSL_AID.utils.TestEasyDict method) (IPSL_AID.utils.TestFileUtils method) setup_data_paths() (in module IPSL_AID.main) setup_directories_and_logging() (in module IPSL_AID.main) setup_model() (in module IPSL_AID.main)

setup_training_environment() (in module IPSL_AID.main) show_header() (IPSL_AID.logger.Logger method) sigma() (IPSL_AID.loss.VPLoss method), [1] (IPSL_AID.networks.VPPrecond method) sigma_data (IPSL_AID.loss.EDMLoss attribute) (IPSL_AID.networks.EDMPrecond attribute) sigma_inv() (IPSL_AID.networks.VPPrecond method) sigma_max (IPSL_AID.loss.VELoss attribute) (IPSL_AID.networks.EDMPrecond attribute) (IPSL_AID.networks.iDDPMPrecond attribute) (IPSL_AID.networks.VEPrecond attribute) (IPSL_AID.networks.VPPrecond attribute) sigma_min (IPSL_AID.loss.VELoss attribute) (IPSL_AID.networks.EDMPrecond attribute) (IPSL_AID.networks.iDDPMPrecond attribute) (IPSL_AID.networks.VEPrecond attribute) (IPSL_AID.networks.VPPrecond attribute) skip (IPSL_AID.networks.UNetBlock attribute) SongUNet (class in IPSL_AID.networks) spread_skill_ratio() (in module IPSL_AID.diagnostics) start_task() (IPSL_AID.logger.Logger method) stats() (in module IPSL_AID.dataset) step() (IPSL_AID.logger.Logger method) success() (IPSL_AID.logger.Logger method)

T

tearDown() (IPSL_AID.dataset.TestDataPreprocessor method) (IPSL_AID.diagnostics.TestPlottingFunctions method) (IPSL_AID.evaluater.TestRunValidation method) (IPSL_AID.logger.TestLogger method) (IPSL_AID.loss.TestLosses method) (IPSL_AID.model.TestModelLoader method) (IPSL_AID.model_utils.TestModelUtils method) (IPSL_AID.networks.TestDiffusionNetworks method) (IPSL_AID.utils.TestEasyDict method) (IPSL_AID.utils.TestFileUtils method) test_adm_edm_combination() (IPSL_AID.model.TestModelLoader method) test_adm_unet_combination() (IPSL_AID.model.TestModelLoader method) test_attribute_error_for_nonexistent() (IPSL_AID.utils.TestEasyDict method) test_attribute_get_set() (IPSL_AID.utils.TestEasyDict method) test_basic() (IPSL_AID.evaluater.TestErrorMetrics method) test_clear_logs_with_record() (IPSL_AID.logger.TestLogger method) test_clear_logs_without_record() (IPSL_AID.logger.TestLogger method) test_coarse_down_up_with_numpy_arrays() (IPSL_AID.dataset.TestDataPreprocessor method) test_coarse_down_up_with_torch_tensors() (IPSL_AID.dataset.TestDataPreprocessor method) test_count_parameters_by_layer() (IPSL_AID.model_utils.TestModelUtils method) test_crps_basic() (IPSL_AID.evaluater.TestCRPSFunction method) test_crps_dtype_preservation() (IPSL_AID.evaluater.TestCRPSFunction method) test_crps_equals_mae_for_single_member() (IPSL_AID.evaluater.TestCRPSFunction method) test_crps_multi_dimensional_flatten() (IPSL_AID.evaluater.TestCRPSFunction method) test_crps_zero_when_perfect_prediction() (IPSL_AID.evaluater.TestCRPSFunction method) test_crps_zero_when_predictions_equal_fine() (IPSL_AID.evaluater.TestRunValidation method) test_curl() (IPSL_AID.diagnostics.TestPlottingFunctions method) test_dataset_len_method() (IPSL_AID.dataset.TestDataPreprocessor method) test_ddpmpp_vp_combination() (IPSL_AID.model.TestModelLoader method) test_delattr() (IPSL_AID.utils.TestEasyDict method) test_delitem() (IPSL_AID.utils.TestEasyDict method) test_denormalize_log1p_minmax() (IPSL_AID.evaluater.TestDenormalizeFunction method) test_denormalize_log1p_standard() (IPSL_AID.evaluater.TestDenormalizeFunction method) test_denormalize_minmax() (IPSL_AID.evaluater.TestDenormalizeFunction method) test_denormalize_minmax_11() (IPSL_AID.evaluater.TestDenormalizeFunction method) test_denormalize_robust() (IPSL_AID.evaluater.TestDenormalizeFunction method) test_denormalize_standard() (IPSL_AID.evaluater.TestDenormalizeFunction method) test_denormalize_unsupported_type() (IPSL_AID.evaluater.TestDenormalizeFunction method) test_denormalize_zero_denominator() (IPSL_AID.evaluater.TestDenormalizeFunction method) test_dhariwal_unet() (IPSL_AID.networks.TestDiffusionNetworks method) test_dict_get_set() (IPSL_AID.utils.TestEasyDict method) test_dict_methods() (IPSL_AID.utils.TestEasyDict method) test_different_shapes() (IPSL_AID.evaluater.TestErrorMetrics method) test_divergence() (IPSL_AID.diagnostics.TestPlottingFunctions method) test_dry_frequency_map() (IPSL_AID.diagnostics.TestPlottingFunctions method) test_dtype_preservation() (IPSL_AID.evaluater.TestErrorMetrics method) test_easydict_with_easydict() (IPSL_AID.utils.TestEasyDict method) test_edm_loss() (IPSL_AID.loss.TestLosses method) test_edm_preconditioner() (IPSL_AID.networks.TestDiffusionNetworks method) test_empty_initialization() (IPSL_AID.utils.TestEasyDict method) test_error_file_output() (IPSL_AID.logger.TestLogger method) test_error_with_exception() (IPSL_AID.logger.TestLogger method) test_error_without_exception() (IPSL_AID.logger.TestLogger method) test_exact_match() (IPSL_AID.evaluater.TestErrorMetrics method) test_example_from_docstring() (IPSL_AID.evaluater.TestErrorMetrics method) test_exception_file_output() (IPSL_AID.logger.TestLogger method) test_exception_with_exception() (IPSL_AID.logger.TestLogger method) test_exception_without_exception() (IPSL_AID.logger.TestLogger method) test_extract_batch() (IPSL_AID.dataset.TestDataPreprocessor method) test_full_checkpoint_cycle() (IPSL_AID.model_utils.TestModelUtils method) test_full_logging_cycle() (IPSL_AID.logger.TestLogger method) test_gaussian_filter_basic() (IPSL_AID.dataset.TestDataPreprocessor method) test_gaussian_filter_different_epsilon() (IPSL_AID.dataset.TestDataPreprocessor method) test_generate_evaluation_slices() (IPSL_AID.dataset.TestDataPreprocessor method) test_generate_random_batch_centers() (IPSL_AID.dataset.TestDataPreprocessor method) test_generate_region_slices() (IPSL_AID.dataset.TestDataPreprocessor method) test_generate_residuals_matches_fine() (IPSL_AID.evaluater.TestRunValidation method) test_get_center_indices_from_latlon() (IPSL_AID.dataset.TestDataPreprocessor method) test_get_parameter_number() (IPSL_AID.model_utils.TestModelUtils method) test_get_parameter_number_with_frozen_layers() (IPSL_AID.model_utils.TestModelUtils method) test_getitem_train_mode() (IPSL_AID.dataset.TestDataPreprocessor method) test_getitem_validation_mode() (IPSL_AID.dataset.TestDataPreprocessor method) test_getitem_with_filter_enabled() (IPSL_AID.dataset.TestDataPreprocessor method) test_info_console() (IPSL_AID.logger.TestLogger method) test_info_file() (IPSL_AID.logger.TestLogger method) test_initialization_console_only() (IPSL_AID.logger.TestLogger method) test_initialization_default() (IPSL_AID.logger.TestLogger method) test_initialization_file_only() (IPSL_AID.logger.TestLogger method) test_initialization_with_dict() (IPSL_AID.utils.TestEasyDict method) test_initialization_with_file_output() (IPSL_AID.logger.TestLogger method) test_initialization_with_kwargs() (IPSL_AID.utils.TestEasyDict method) test_initialization_with_record() (IPSL_AID.logger.TestLogger method) test_invalid_combinations() (IPSL_AID.model.TestModelLoader method) test_invalid_coordinate_handling() (IPSL_AID.dataset.TestDataPreprocessor method) test_invalid_time_normalization() (IPSL_AID.dataset.TestDataPreprocessor method) test_key_error_for_nonexistent() (IPSL_AID.utils.TestEasyDict method) test_kl_different_shapes() (IPSL_AID.evaluater.TestErrorMetrics method) test_kl_divergence_basic() (IPSL_AID.evaluater.TestErrorMetrics method) test_kl_dtype_preservation() (IPSL_AID.evaluater.TestErrorMetrics method) test_kl_multi_dimensional() (IPSL_AID.evaluater.TestErrorMetrics method) test_load_checkpoint() (IPSL_AID.model_utils.TestModelUtils method) test_load_checkpoint_without_optimizer() (IPSL_AID.model_utils.TestModelUtils method) test_load_training_checkpoint() (IPSL_AID.model_utils.TestModelUtils method) test_load_training_checkpoint_nonexistent() (IPSL_AID.model_utils.TestModelUtils method) test_log_metrics_empty() (IPSL_AID.logger.TestLogger method) test_log_metrics_file_output() (IPSL_AID.logger.TestLogger method) test_log_metrics_with_data() (IPSL_AID.logger.TestLogger method) test_log_model_summary_with_input_shape() (IPSL_AID.model_utils.TestModelUtils method) test_log_model_summary_without_input_shape() (IPSL_AID.model_utils.TestModelUtils method) test_logger_with_unicode() (IPSL_AID.logger.TestLogger method) test_loss_comparison() (IPSL_AID.loss.TestLosses method) test_loss_gradients() (IPSL_AID.loss.TestLosses method) test_loss_with_augmentation() (IPSL_AID.loss.TestLosses method) test_makedir_existing_directory() (IPSL_AID.utils.TestFileUtils method)

test_makedir_multiple_nested_directories() (IPSL_AID.utils.TestFileUtils method) test_makedir_new_directory() (IPSL_AID.utils.TestFileUtils method) test_makedir_then_makefile() (IPSL_AID.utils.TestFileUtils method) test_makefile_existing_file() (IPSL_AID.utils.TestFileUtils method) test_makefile_in_nonexistent_directory() (IPSL_AID.utils.TestFileUtils method) test_makefile_multiple_files() (IPSL_AID.utils.TestFileUtils method) test_makefile_new_file() (IPSL_AID.utils.TestFileUtils method) test_metric_plots_comprehensive() (IPSL_AID.diagnostics.TestPlottingFunctions method) test_metric_tracker_example_from_docstring() (IPSL_AID.evaluater.TestMetricTracker method) test_metric_tracker_getmean() (IPSL_AID.evaluater.TestMetricTracker method) test_metric_tracker_getsqrtmean() (IPSL_AID.evaluater.TestMetricTracker method) test_metric_tracker_getstd() (IPSL_AID.evaluater.TestMetricTracker method) test_metric_tracker_init() (IPSL_AID.evaluater.TestMetricTracker method) test_metric_tracker_reset() (IPSL_AID.evaluater.TestMetricTracker method) test_metric_tracker_update() (IPSL_AID.evaluater.TestMetricTracker method) test_mixed_access() (IPSL_AID.utils.TestEasyDict method) test_model_kwargs_override() (IPSL_AID.model.TestModelLoader method) test_multi_dimensional() (IPSL_AID.evaluater.TestErrorMetrics method) test_mv_correlation() (IPSL_AID.diagnostics.TestPlottingFunctions method) test_ncsnpp_ve_combination() (IPSL_AID.model.TestModelLoader method) test_nested_easydict() (IPSL_AID.utils.TestEasyDict method) test_new_epoch_method() (IPSL_AID.dataset.TestDataPreprocessor method) test_no_conditional_channels() (IPSL_AID.model.TestModelLoader method) test_normalize_methods() (IPSL_AID.dataset.TestDataPreprocessor method) test_normalize_time_cos_sin() (IPSL_AID.dataset.TestDataPreprocessor method) test_normalize_time_linear() (IPSL_AID.dataset.TestDataPreprocessor method) test_parameter_counts() (IPSL_AID.networks.TestDiffusionNetworks method) test_plot_dry_frequency_map_comprehensive() (IPSL_AID.diagnostics.TestPlottingFunctions method) test_plot_ensemble_surface_comprehensive() (IPSL_AID.diagnostics.TestPlottingFunctions method) test_plot_error_map_comprehensive() (IPSL_AID.diagnostics.TestPlottingFunctions method) test_plot_global_surface_robinson_comprehensive() (IPSL_AID.diagnostics.TestPlottingFunctions method) test_plot_mae_map_comprehensive() (IPSL_AID.diagnostics.TestPlottingFunctions method) test_plot_mean_curl_map_comprehensive() (IPSL_AID.diagnostics.TestPlottingFunctions method) test_plot_mean_divergence_map_comprehensive() (IPSL_AID.diagnostics.TestPlottingFunctions method) test_plot_metrics_heatmap_comprehensive() (IPSL_AID.diagnostics.TestPlottingFunctions method) test_plot_ranks() (IPSL_AID.diagnostics.TestPlottingFunctions method) test_plot_spread_skill_ratio_hexbin_comprehensive() (IPSL_AID.diagnostics.TestPlottingFunctions method) test_plot_spread_skill_ratio_map_comprehensive() (IPSL_AID.diagnostics.TestPlottingFunctions method) test_plot_surface_comprehensive() (IPSL_AID.diagnostics.TestPlottingFunctions method) test_plot_zoom_comparison_comprehensive() (IPSL_AID.diagnostics.TestPlottingFunctions method) test_power_spectra_comprehensive() (IPSL_AID.diagnostics.TestPlottingFunctions method) test_preprocessor_batch_size_validation() (IPSL_AID.dataset.TestDataPreprocessor method) test_preprocessor_initialization_cos_sin_normalization() (IPSL_AID.dataset.TestDataPreprocessor method) test_preprocessor_initialization_train_mode() (IPSL_AID.dataset.TestDataPreprocessor method) test_preprocessor_initialization_validation_mode() (IPSL_AID.dataset.TestDataPreprocessor method) test_preprocessor_without_constants() (IPSL_AID.dataset.TestDataPreprocessor method) test_print_model_layers() (IPSL_AID.model_utils.TestModelUtils method) test_progress_bar() (IPSL_AID.logger.TestLogger method) test_qq_quantiles_comprehensive() (IPSL_AID.diagnostics.TestPlottingFunctions method) test_ranks() (IPSL_AID.diagnostics.TestPlottingFunctions method) test_rectangular_resolution() (IPSL_AID.model.TestModelLoader method) test_save_checkpoint() (IPSL_AID.model_utils.TestModelUtils method) test_save_emergency_checkpoint() (IPSL_AID.model_utils.TestModelUtils method) test_save_training_checkpoint_best_type() (IPSL_AID.model_utils.TestModelUtils method) test_save_training_checkpoint_epoch_type() (IPSL_AID.model_utils.TestModelUtils method) test_save_training_checkpoint_final_type() (IPSL_AID.model_utils.TestModelUtils method) test_save_training_checkpoint_samples_type() (IPSL_AID.model_utils.TestModelUtils method) test_show_header_both() (IPSL_AID.logger.TestLogger method) test_show_header_console() (IPSL_AID.logger.TestLogger method) test_show_header_file() (IPSL_AID.logger.TestLogger method) test_song_unet_rectangular_resolution() (IPSL_AID.networks.TestDiffusionNetworks method) test_song_unet_square_resolution() (IPSL_AID.networks.TestDiffusionNetworks method) test_spatiotemporal_histograms_comprehensive() (IPSL_AID.diagnostics.TestPlottingFunctions method) test_ssr_basic() (IPSL_AID.diagnostics.TestSSRFunction method) test_ssr_one_when_perfect_prediction() (IPSL_AID.diagnostics.TestSSRFunction method) test_start_task_file_output() (IPSL_AID.logger.TestLogger method) test_start_task_minimal() (IPSL_AID.logger.TestLogger method) test_start_task_with_description() (IPSL_AID.logger.TestLogger method) test_start_task_with_metadata() (IPSL_AID.logger.TestLogger method) test_stats_computation_with_existing_json() (IPSL_AID.dataset.TestDataPreprocessor method) test_stats_computation_without_existing_json() (IPSL_AID.dataset.TestDataPreprocessor method) test_step_console() (IPSL_AID.logger.TestLogger method) test_step_file() (IPSL_AID.logger.TestLogger method) test_success_console() (IPSL_AID.logger.TestLogger method) test_success_file() (IPSL_AID.logger.TestLogger method) test_temporal_series_comparison_comprehensive() (IPSL_AID.diagnostics.TestPlottingFunctions method) test_unet_loss() (IPSL_AID.loss.TestLosses method) test_val_loss_and_metrics_across_3_batches_consistent_shape() (IPSL_AID.evaluater.TestRunValidation method) test_validation_hexbin_comprehensive() (IPSL_AID.diagnostics.TestPlottingFunctions method) test_validation_pdfs_comprehensive() (IPSL_AID.diagnostics.TestPlottingFunctions method) test_ve_loss() (IPSL_AID.loss.TestLosses method) test_ve_preconditioner() (IPSL_AID.networks.TestDiffusionNetworks method) test_vp_loss() (IPSL_AID.loss.TestLosses method) test_vp_preconditioner() (IPSL_AID.networks.TestDiffusionNetworks method) test_warning_console() (IPSL_AID.logger.TestLogger method) test_warning_file() (IPSL_AID.logger.TestLogger method) TestCRPSFunction (class in IPSL_AID.evaluater) TestDataPreprocessor (class in IPSL_AID.dataset) TestDenormalizeFunction (class in IPSL_AID.evaluater) TestDiffusionNetworks (class in IPSL_AID.networks) TestEasyDict (class in IPSL_AID.utils) TestErrorMetrics (class in IPSL_AID.evaluater) TestFileUtils (class in IPSL_AID.utils) TestLogger (class in IPSL_AID.logger) TestLosses (class in IPSL_AID.loss) TestMetricTracker (class in IPSL_AID.evaluater) TestModel (class in IPSL_AID.model_utils) TestModelLoader (class in IPSL_AID.model) TestModelUtils (class in IPSL_AID.model_utils) TestPlottingFunctions (class in IPSL_AID.diagnostics) TestRunValidation (class in IPSL_AID.evaluater) TestSSRFunction (class in IPSL_AID.diagnostics) time (IPSL_AID.dataset.DataPreprocessor attribute) time_batchs (IPSL_AID.dataset.DataPreprocessor attribute) tindex_tracker (IPSL_AID.dataset.DataPreprocessor attribute)

U

u (IPSL_AID.networks.iDDPMPrecond attribute) UNetBlock (class in IPSL_AID.networks) UnetLoss (class in IPSL_AID.loss) update() (IPSL_AID.evaluater.MetricTracker method)

use_fp16 (IPSL_AID.networks.EDMPrecond attribute) (IPSL_AID.networks.iDDPMPrecond attribute) (IPSL_AID.networks.VEPrecond attribute) (IPSL_AID.networks.VPPrecond attribute)

V

value (IPSL_AID.evaluater.MetricTracker attribute) VELoss (class in IPSL_AID.loss)

VEPrecond (class in IPSL_AID.networks) VPLoss (class in IPSL_AID.loss) VPPrecond (class in IPSL_AID.networks)

W

warning() (IPSL_AID.logger.Logger method) weight (IPSL_AID.networks.Conv2d attribute) (IPSL_AID.networks.GroupNorm attribute) (IPSL_AID.networks.Linear attribute)

weight_init() (in module IPSL_AID.networks)

Y

year (IPSL_AID.dataset.DataPreprocessor attribute)

year_norm (IPSL_AID.dataset.DataPreprocessor attribute)