Index

_

__call__() (IPSL_AID.loss.EDMLoss method) (IPSL_AID.loss.VELoss method) __init__() (AID_BC.bias_corrector.BiasCorrector method) (AID_BC.bias_corrector.OptimalTransportCorrector method) (AID_BC.bias_corrector.QuantileMappingCorrector method) (AID_BC.dataset.ClimateDataset method) (AID_BC.evaluater.MetricTracker method) (AID_BC.logger.Logger method) (AID_BC.optimal_transport.OptimalTransport method) (AID_BC.quantile_mapping.QM method) (IPSL_AID.dataset.DataPreprocessor method) (IPSL_AID.evaluater.MetricTracker method) (IPSL_AID.logger.Logger method) (IPSL_AID.loss.EDMLoss method) (IPSL_AID.loss.UnetLoss method) (IPSL_AID.loss.VELoss method) (IPSL_AID.loss.VPLoss method) (IPSL_AID.model_utils.ModelUtils 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) (IPSL_AID.utils.FileUtils method)

A

affine (IPSL_AID.networks.UNetBlock attribute) AID_BC.bias_corrector module AID_BC.dataset module AID_BC.diagnostics module AID_BC.evaluater module AID_BC.logger module AID_BC.main module

AID_BC.optimal_transport module AID_BC.preprocess module AID_BC.quantile_mapping module AID_BC.version module alpha_bar() (IPSL_AID.networks.iDDPMPrecond method) apply_bias_corrector() (in module AID_BC.main) apply_bias_corrector_by_month() (in module AID_BC.main) assemble_variable_dataset() (in module AID_BC.main) 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) BiasCorrector (class in AID_BC.bias_corrector)

BORDER_STYLE (AID_BC.diagnostics.PlotConfig attribute) (IPSL_AID.diagnostics.PlotConfig attribute) BORDER_w (AID_BC.diagnostics.PlotConfig attribute) (IPSL_AID.diagnostics.PlotConfig attribute) build_fine_coarse_blocks() (IPSL_AID.dataset.DataPreprocessor method) build_path() (in module AID_BC.main) (in module AID_BC.preprocess)

C

calculate_pearsoncorr_nparray() (in module IPSL_AID.diagnostics) calculate_psd2d_simple() (in module AID_BC.diagnostics) (in module IPSL_AID.diagnostics) center (AID_BC.bias_corrector.OptimalTransportCorrector attribute) center_tracker (IPSL_AID.dataset.DataPreprocessor attribute) check_dataset_consistency() (in module AID_BC.main) check_same_spatial_grid() (in module AID_BC.main) clear_logs() (AID_BC.logger.Logger method) (IPSL_AID.logger.Logger method) ClimateDataset (class in AID_BC.dataset) close() (AID_BC.dataset.ClimateDataset method) cmip6_data (AID_BC.dataset.ClimateDataset attribute) coarse_down_up() (in module IPSL_AID.dataset) COASTLINE_w (AID_BC.diagnostics.PlotConfig attribute) (IPSL_AID.diagnostics.PlotConfig attribute) COLORBAR_h (AID_BC.diagnostics.PlotConfig attribute) (IPSL_AID.diagnostics.PlotConfig attribute) COLORBAR_PAD (AID_BC.diagnostics.PlotConfig attribute) (IPSL_AID.diagnostics.PlotConfig attribute)

COLORMAPS (AID_BC.diagnostics.PlotConfig attribute) (IPSL_AID.diagnostics.PlotConfig attribute) compute_sal() (in module IPSL_AID.diagnostics) compute_sal_objects() (in module IPSL_AID.diagnostics) compute_sal_objects_time() (in module IPSL_AID.diagnostics) const_vars (IPSL_AID.dataset.DataPreprocessor attribute) Conv2d (class in IPSL_AID.networks) converged (AID_BC.bias_corrector.OptimalTransportCorrector attribute) (AID_BC.optimal_transport.TransportSolution attribute) convert_units() (AID_BC.diagnostics.PlotConfig class method) (IPSL_AID.diagnostics.PlotConfig class method) cost (AID_BC.optimal_transport.TransportSolution property) cost_matrix (AID_BC.optimal_transport.TransportSolution attribute) count (IPSL_AID.evaluater.MetricTracker attribute) count_parameters_by_layer() (IPSL_AID.model_utils.ModelUtils static method) create_bias_corrector() (in module AID_BC.bias_corrector) create_data_loaders() (in module IPSL_AID.main) 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 (AID_BC.diagnostics.PlotConfig attribute) (IPSL_AID.diagnostics.PlotConfig attribute) DEFAULT_SAVE_DIR (AID_BC.diagnostics.PlotConfig attribute) (IPSL_AID.diagnostics.PlotConfig attribute)

denormalize() (in module IPSL_AID.evaluater) DhariwalUNet (class in IPSL_AID.networks) diagnostics (AID_BC.bias_corrector.BiasCorrector property) (AID_BC.bias_corrector.OptimalTransportCorrector property) 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 (AID_BC.optimal_transport.TransportSolution attribute) epsilon_t (IPSL_AID.loss.VPLoss attribute)

era5_data (AID_BC.dataset.ClimateDataset attribute) error() (AID_BC.logger.Logger method) (IPSL_AID.logger.Logger method) eval_region_indices (IPSL_AID.dataset.DataPreprocessor attribute) eval_slices (IPSL_AID.dataset.DataPreprocessor attribute) evaluate() (in module AID_BC.evaluater) exception() (AID_BC.logger.Logger method) (IPSL_AID.logger.Logger method) extract_batch() (IPSL_AID.dataset.DataPreprocessor method)

F

FileUtils (class in IPSL_AID.utils) filter_batch() (IPSL_AID.dataset.DataPreprocessor method) fit() (AID_BC.bias_corrector.BiasCorrector method) (AID_BC.bias_corrector.OptimalTransportCorrector method) (AID_BC.bias_corrector.QuantileMappingCorrector method) (AID_BC.quantile_mapping.QM method) fit_bias_corrector() (in module AID_BC.main) fit_bias_correctors_by_month() (in module AID_BC.main) fit_transform() (AID_BC.bias_corrector.BiasCorrector method) fitted_iterations (AID_BC.bias_corrector.OptimalTransportCorrector attribute) FIXED_DIFF_RANGES (AID_BC.diagnostics.PlotConfig attribute) (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) flatten_dataset() (in module AID_BC.main)

forward() (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) fu (AID_BC.optimal_transport.TransportSolution property)

G

gaussian_filter() (in module IPSL_AID.dataset) generate_evaluation_slices() (IPSL_AID.dataset.DataPreprocessor method) generate_random_batch_centers() (IPSL_AID.dataset.DataPreprocessor method) 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() (AID_BC.diagnostics.PlotConfig class method) (IPSL_AID.diagnostics.PlotConfig class method) get_curl() (in module IPSL_AID.diagnostics) get_divergence() (in module IPSL_AID.diagnostics) get_fixed_diff_range() (AID_BC.diagnostics.PlotConfig static method) (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() (AID_BC.diagnostics.PlotConfig class method) (IPSL_AID.diagnostics.PlotConfig class method) get_version() (in module AID_BC.version) getmax() (AID_BC.evaluater.MetricTracker method) getmean() (AID_BC.evaluater.MetricTracker method) (IPSL_AID.evaluater.MetricTracker method) getmin() (AID_BC.evaluater.MetricTracker method) getsqrtmean() (AID_BC.evaluater.MetricTracker method) (IPSL_AID.evaluater.MetricTracker method) getstd() (AID_BC.evaluater.MetricTracker method) (IPSL_AID.evaluater.MetricTracker method) GroupNorm (class in IPSL_AID.networks) gv (AID_BC.optimal_transport.TransportSolution property)

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() (AID_BC.logger.Logger method) (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 is_fitted (AID_BC.bias_corrector.BiasCorrector attribute) is_latitude_descending() (AID_BC.dataset.ClimateDataset static method)

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 (AID_BC.diagnostics.PlotConfig attribute) (IPSL_AID.diagnostics.PlotConfig attribute) Linear (class in IPSL_AID.networks) load_application_data() (in module AID_BC.main) load_checkpoint() (IPSL_AID.model_utils.ModelUtils static method) load_cmip6() (AID_BC.dataset.ClimateDataset method) load_dynamic_covariates() (IPSL_AID.dataset.DataPreprocessor method) load_era5() (AID_BC.dataset.ClimateDataset method)

load_features() (IPSL_AID.dataset.DataPreprocessor method) load_model_and_loss() (in module IPSL_AID.model) load_precomputed_metadata() (in module AID_BC.main) load_training_checkpoint() (IPSL_AID.model_utils.ModelUtils static method) load_training_data() (in module AID_BC.main) load_training_variable() (in module AID_BC.main) log_configuration() (in module IPSL_AID.main) log_metrics() (AID_BC.logger.Logger method) (IPSL_AID.logger.Logger method) log_model_summary() (IPSL_AID.model_utils.ModelUtils static method) Logger (class in AID_BC.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 AID_BC.main) (in module AID_BC.preprocess) (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) make_starts() (in module IPSL_AID.dataset) 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) method_name (AID_BC.bias_corrector.BiasCorrector attribute) (AID_BC.bias_corrector.OptimalTransportCorrector attribute) (AID_BC.bias_corrector.QuantileMappingCorrector attribute) MetricTracker (class in AID_BC.evaluater) (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 AID_BC.bias_corrector AID_BC.dataset AID_BC.diagnostics AID_BC.evaluater AID_BC.logger AID_BC.main AID_BC.optimal_transport AID_BC.preprocess AID_BC.quantile_mapping AID_BC.version 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

n_features (AID_BC.bias_corrector.BiasCorrector attribute) new_epoch() (IPSL_AID.dataset.DataPreprocessor method) nmae_all() (in module IPSL_AID.evaluater) normalize() (IPSL_AID.dataset.DataPreprocessor method)

normalize_longitudes() (AID_BC.dataset.ClimateDataset static method) normalize_time() (IPSL_AID.dataset.DataPreprocessor method) num_channels (IPSL_AID.networks.PositionalEmbedding attribute) num_iterations (AID_BC.optimal_transport.TransportSolution attribute)

O

open_dataarray() (in module AID_BC.main) OptimalTransport (class in AID_BC.optimal_transport) OptimalTransportCorrector (class in AID_BC.bias_corrector) 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) overlap_aware_1d_window() (in module IPSL_AID.evaluater) overlap_aware_2d_window() (in module IPSL_AID.evaluater) overlap_ratio (IPSL_AID.dataset.DataPreprocessor attribute)

P

P_mean (IPSL_AID.loss.EDMLoss attribute) P_std (IPSL_AID.loss.EDMLoss attribute) parse_args() (in module AID_BC.main) (in module AID_BC.preprocess) (in module IPSL_AID.download_ERA5_cds) (in module IPSL_AID.generate_all_data_ERA5) (in module IPSL_AID.main) parse_variable_paths() (in module AID_BC.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 AID_BC.diagnostics) (in module IPSL_AID.diagnostics) plot_power_spectra() (in module AID_BC.diagnostics) (in module IPSL_AID.diagnostics) plot_qq_quantiles() (in module AID_BC.diagnostics) (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 AID_BC.diagnostics) (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 AID_BC.diagnostics) (in module IPSL_AID.diagnostics) plot_validation_salpdfs() (in module IPSL_AID.diagnostics) plot_zoom_comparison() (in module IPSL_AID.diagnostics) PlotConfig (class in AID_BC.diagnostics) (class in IPSL_AID.diagnostics) PositionalEmbedding (class in IPSL_AID.networks) potentials (AID_BC.optimal_transport.TransportSolution attribute) predict() (AID_BC.quantile_mapping.QM method) prepare_dataarray() (in module AID_BC.main) prepare_dataset() (AID_BC.dataset.ClimateDataset method) print_metrics() (in module AID_BC.evaluater) print_model_layers() (IPSL_AID.model_utils.ModelUtils static method) process_year() (in module AID_BC.preprocess)

Q

QM (class in AID_BC.quantile_mapping)

QuantileMappingCorrector (class in AID_BC.bias_corrector)

R

r2_all() (in module IPSL_AID.evaluater) radial_average_psd() (in module AID_BC.diagnostics) (in module IPSL_AID.diagnostics) random_centers (IPSL_AID.dataset.DataPreprocessor attribute) ranks() (in module IPSL_AID.diagnostics) reconstruct_dataset() (in module AID_BC.main) reconstruct_original_layout() (in module IPSL_AID.evaluater) reduction (IPSL_AID.loss.UnetLoss attribute) reg_ot_cost (AID_BC.optimal_transport.TransportSolution attribute) region_center (IPSL_AID.dataset.DataPreprocessor attribute) region_size (IPSL_AID.dataset.DataPreprocessor attribute) regularized_ot_cost (AID_BC.bias_corrector.OptimalTransportCorrector attribute)

rename_coordinates() (AID_BC.dataset.ClimateDataset static method) resample_filter (IPSL_AID.networks.Conv2d attribute) reset() (AID_BC.evaluater.MetricTracker method) (IPSL_AID.evaluater.MetricTracker method) resize_with_torchvision() (in module AID_BC.dataset) resolve_precomputed_variable_root() (in module AID_BC.main) 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) sample_time_steps_by_doy() (IPSL_AID.dataset.DataPreprocessor method) sampler() (in module IPSL_AID.evaluater) save_checkpoint() (IPSL_AID.model_utils.ModelUtils static method) save_corrected_year() (in module AID_BC.main) save_emergency_checkpoint() (IPSL_AID.model_utils.ModelUtils static method) save_training_checkpoint() (IPSL_AID.model_utils.ModelUtils static method) scale (AID_BC.bias_corrector.OptimalTransportCorrector attribute) select_month() (in module AID_BC.main) 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() (AID_BC.logger.Logger method) (IPSL_AID.logger.Logger method) sigma() (IPSL_AID.loss.VPLoss method) (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) solve() (AID_BC.optimal_transport.OptimalTransport method) solver (AID_BC.bias_corrector.OptimalTransportCorrector attribute) SongUNet (class in IPSL_AID.networks) sort_latitude() (AID_BC.dataset.ClimateDataset static method) spread_skill_ratio() (in module IPSL_AID.diagnostics) standardize_coordinates() (AID_BC.dataset.ClimateDataset static method) start_task() (AID_BC.logger.Logger method) (IPSL_AID.logger.Logger method) stats() (in module IPSL_AID.dataset) step() (AID_BC.logger.Logger method) (IPSL_AID.logger.Logger method) success() (AID_BC.logger.Logger method) (IPSL_AID.logger.Logger method) summary_metrics() (in module AID_BC.evaluater)

T

target_points (AID_BC.bias_corrector.OptimalTransportCorrector attribute) target_potential (AID_BC.bias_corrector.OptimalTransportCorrector attribute) threshold (AID_BC.optimal_transport.TransportSolution attribute) time (IPSL_AID.dataset.DataPreprocessor attribute) time_batchs (IPSL_AID.dataset.DataPreprocessor attribute) tindex_tracker (IPSL_AID.dataset.DataPreprocessor attribute) to_numpy_4d() (in module AID_BC.diagnostics) train_slices (IPSL_AID.dataset.DataPreprocessor attribute)

transform() (AID_BC.bias_corrector.BiasCorrector method) (AID_BC.bias_corrector.OptimalTransportCorrector method) (AID_BC.bias_corrector.QuantileMappingCorrector method) transport_fn() (AID_BC.optimal_transport.OptimalTransport method) transport_fn_direct() (AID_BC.optimal_transport.OptimalTransport method) transport_function (AID_BC.bias_corrector.OptimalTransportCorrector attribute) transport_plan (AID_BC.optimal_transport.TransportSolution property) TransportSolution (class in AID_BC.optimal_transport)

U

u (IPSL_AID.networks.iDDPMPrecond attribute) UNetBlock (class in IPSL_AID.networks) UnetLoss (class in IPSL_AID.loss) update() (AID_BC.evaluater.MetricTracker method) (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

valid_multiregional_centers() (IPSL_AID.dataset.DataPreprocessor method) validate_args() (in module AID_BC.main) validate_fit_arrays() (AID_BC.bias_corrector.BiasCorrector static method) validate_transform_array() (AID_BC.bias_corrector.BiasCorrector method) validate_variable_paths() (in module AID_BC.main)

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() (AID_BC.logger.Logger method) (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)