IPSL_AID

• IPSL_AID package
  • Submodules
  • IPSL_AID.dataset module
    • stats()
    • coarse_down_up()
    • gaussian_filter()
    • DataPreprocessor
      • DataPreprocessor.const_vars
      • DataPreprocessor.time
      • DataPreprocessor.year
      • DataPreprocessor.month
      • DataPreprocessor.day
      • DataPreprocessor.hour
      • DataPreprocessor.year_norm
      • DataPreprocessor.doy_norm
      • DataPreprocessor.hour_norm
      • DataPreprocessor.time_batchs
      • DataPreprocessor.eval_slices
      • DataPreprocessor.random_centers
      • DataPreprocessor.center_tracker
      • DataPreprocessor.tindex_tracker
      • DataPreprocessor.new_epoch()
      • DataPreprocessor.sample_time_steps_by_doy()
      • DataPreprocessor.sample_random_time_indices()
      • DataPreprocessor.load_dynamic_covariates()
      • DataPreprocessor.generate_random_batch_centers()
      • DataPreprocessor.generate_evaluation_slices()
      • DataPreprocessor.extract_batch()
      • DataPreprocessor.filter_batch()
      • DataPreprocessor.normalize()
      • DataPreprocessor.normalize_time()
      • DataPreprocessor.__len__()
      • DataPreprocessor.__getitem__()
      • DataPreprocessor.__init__()
      • DataPreprocessor.new_epoch()
      • DataPreprocessor.sample_time_steps_by_doy()
      • DataPreprocessor.sample_random_time_indices()
      • DataPreprocessor.load_dynamic_covariates()
      • DataPreprocessor.get_center_indices_from_latlon()
      • DataPreprocessor.generate_random_batch_centers()
      • DataPreprocessor.generate_evaluation_slices()
      • DataPreprocessor.generate_region_slices()
      • DataPreprocessor.extract_batch()
      • DataPreprocessor.build_fine_coarse_blocks()
      • DataPreprocessor.filter_batch()
      • DataPreprocessor.normalize()
      • DataPreprocessor.normalize_time()
  • IPSL_AID.diagnostics module
    • PlotConfig
      • PlotConfig.DEFAULT_SAVE_DIR
      • PlotConfig.DEFAULT_FIGSIZE_MULTIPLIER
      • PlotConfig.COLORMAPS
      • PlotConfig.FIXED_DIFF_RANGES
      • PlotConfig.FIXED_DIFF_RANGES_ERRORS
      • PlotConfig.FIXED_MAE_RANGES
      • PlotConfig.FIXED_SSR_RANGES
      • PlotConfig.COASTLINE_w
      • PlotConfig.BORDER_w
      • PlotConfig.LAKE_w
      • PlotConfig.BORDER_STYLE
      • PlotConfig.COLORBAR_h
      • PlotConfig.COLORBAR_PAD
      • PlotConfig.get_colormap()
      • PlotConfig.get_plot_name()
      • PlotConfig.convert_units()
      • PlotConfig.get_fixed_diff_range()
      • PlotConfig.get_fixed_diff_range_errors()
      • PlotConfig.get_fixed_mae_range()
      • PlotConfig.get_fixed_ssr_range()
    • plot_validation_hexbin()
    • plot_comparison_hexbin()
    • plot_metric_histories()
    • plot_metrics_heatmap()
    • plot_loss_histories()
    • plot_average_metrics()
    • plot_spatiotemporal_histograms()
    • plot_surface()
    • plot_ensemble_surface()
    • plot_zoom_comparison()
    • plot_global_surface_robinson()
    • plot_MAE_map()
    • plot_error_map()
    • spread_skill_ratio()
    • plot_spread_skill_ratio_map()
    • plot_spread_skill_ratio_hexbin()
    • plot_validation_pdfs()
    • plot_power_spectra()
    • calculate_psd2d_simple()
    • radial_average_psd()
    • plot_qq_quantiles()
    • dry_frequency_map()
    • plot_dry_frequency_map()
    • calculate_pearsoncorr_nparray()
    • plot_validation_mvcorr_space()
    • plot_validation_mvcorr()
    • plot_temporal_series_comparison()
    • ranks()
    • plot_ranks()
    • get_divergence()
    • get_curl()
    • plot_mean_divergence_map()
    • plot_mean_curl_map()
  • IPSL_AID.evaluater module
    • MetricTracker
      • MetricTracker.value
      • MetricTracker.count
      • MetricTracker.__init__()
      • MetricTracker.reset()
      • MetricTracker.update()
      • MetricTracker.getmean()
      • MetricTracker.getstd()
      • MetricTracker.getsqrtmean()
    • mae_all()
    • nmae_all()
    • crps_ensemble_all()
    • rmse_all()
    • r2_all()
    • pearson_all()
    • kl_divergence_all()
    • denormalize()
    • edm_sampler()
    • sampler()
    • reconstruct_original_layout()
    • generate_residuals_norm()
    • run_validation()
  • IPSL_AID.logger module
    • Logger
      • Logger.__init__()
      • Logger.clear_logs()
      • Logger.show_header()
      • Logger.start_task()
      • Logger.log_metrics()
      • Logger.info()
      • Logger.warning()
      • Logger.success()
      • Logger.step()
      • Logger.exception()
      • Logger.error()
  • IPSL_AID.loss module
    • VPLoss
      • VPLoss.beta_d
      • VPLoss.beta_min
      • VPLoss.epsilon_t
      • VPLoss.__call__()
      • VPLoss.sigma()
      • VPLoss.__init__()
      • VPLoss.sigma()
    • VELoss
      • VELoss.sigma_min
      • VELoss.sigma_max
      • VELoss.__call__()
      • VELoss.__init__()
    • EDMLoss
      • EDMLoss.P_mean
      • EDMLoss.P_std
      • EDMLoss.sigma_data
      • EDMLoss.__call__()
      • EDMLoss.__init__()
    • UnetLoss
      • UnetLoss.loss_type
      • UnetLoss.reduction
      • UnetLoss.loss_fn
      • UnetLoss.__init__()
  • IPSL_AID.main module
    • parse_args()
    • make_divisible_hw()
    • setup_directories_and_logging()
    • log_configuration()
    • setup_data_paths()
    • setup_training_environment()
    • create_data_loaders()
    • setup_model()
    • resolve_region_center()
    • main()
  • IPSL_AID.model module
    • load_model_and_loss()
  • IPSL_AID.model_utils module
    • ModelUtils
      • ModelUtils.__init__()
      • ModelUtils.get_parameter_number()
      • ModelUtils.print_model_layers()
      • ModelUtils.save_checkpoint()
      • ModelUtils.load_checkpoint()
      • ModelUtils.load_training_checkpoint()
      • ModelUtils.count_parameters_by_layer()
      • ModelUtils.log_model_summary()
      • ModelUtils.save_training_checkpoint()
      • ModelUtils.save_emergency_checkpoint()
  • IPSL_AID.networks module
    • weight_init()
    • Linear
      • Linear.weight
      • Linear.bias
      • Linear.__init__()
      • Linear.forward()
    • Conv2d
      • Conv2d.weight
      • Conv2d.bias
      • Conv2d.resample_filter
      • Conv2d.__init__()
      • Conv2d.forward()
    • GroupNorm
      • GroupNorm.weight
      • GroupNorm.bias
      • GroupNorm.__init__()
      • GroupNorm.forward()
    • AttentionOp
      • AttentionOp.forward()
      • AttentionOp.backward()
    • UNetBlock
      • UNetBlock.affine
      • UNetBlock.skip
      • UNetBlock.__init__()
      • UNetBlock.forward()
    • PositionalEmbedding
      • PositionalEmbedding.num_channels
      • PositionalEmbedding.max_positions
      • PositionalEmbedding.endpoint
      • PositionalEmbedding.__init__()
      • PositionalEmbedding.forward()
    • FourierEmbedding
      • FourierEmbedding.freqs
      • FourierEmbedding.__init__()
      • FourierEmbedding.forward()
    • SongUNet
      • SongUNet.img_resolution
      • SongUNet.img_height
      • SongUNet.img_width
      • SongUNet.label_dropout
      • SongUNet.map_noise
      • SongUNet.map_label
      • SongUNet.map_augment
      • SongUNet.enc
      • SongUNet.dec
      • SongUNet.__init__()
      • SongUNet.forward()
    • DhariwalUNet
      • DhariwalUNet.img_resolution
      • DhariwalUNet.img_height
      • DhariwalUNet.img_width
      • DhariwalUNet.label_dropout
      • DhariwalUNet.map_noise
      • DhariwalUNet.map_label
      • DhariwalUNet.map_augment
      • DhariwalUNet.enc
      • DhariwalUNet.dec
      • DhariwalUNet.out_norm
      • DhariwalUNet.out_conv
      • DhariwalUNet.__init__()
      • DhariwalUNet.forward()
    • VPPrecond
      • VPPrecond.img_resolution
      • VPPrecond.in_channels
      • VPPrecond.out_channels
      • VPPrecond.label_dim
      • VPPrecond.use_fp16
      • VPPrecond.sigma_min
      • VPPrecond.sigma_max
      • VPPrecond.model
      • VPPrecond.__init__()
      • VPPrecond.forward()
      • VPPrecond.sigma()
      • VPPrecond.sigma_inv()
      • VPPrecond.round_sigma()
    • VEPrecond
      • VEPrecond.img_resolution
      • VEPrecond.in_channels
      • VEPrecond.out_channels
      • VEPrecond.label_dim
      • VEPrecond.use_fp16
      • VEPrecond.sigma_min
      • VEPrecond.sigma_max
      • VEPrecond.model
      • VEPrecond.__init__()
      • VEPrecond.forward()
      • VEPrecond.round_sigma()
    • iDDPMPrecond
      • iDDPMPrecond.img_resolution
      • iDDPMPrecond.in_channels
      • iDDPMPrecond.out_channels
      • iDDPMPrecond.label_dim
      • iDDPMPrecond.use_fp16
      • iDDPMPrecond.sigma_min
      • iDDPMPrecond.sigma_max
      • iDDPMPrecond.u
      • iDDPMPrecond.model
      • iDDPMPrecond.__init__()
      • iDDPMPrecond.forward()
      • iDDPMPrecond.alpha_bar()
      • iDDPMPrecond.round_sigma()
    • EDMPrecond
      • EDMPrecond.img_resolution
      • EDMPrecond.in_channels
      • EDMPrecond.out_channels
      • EDMPrecond.label_dim
      • EDMPrecond.use_fp16
      • EDMPrecond.sigma_min
      • EDMPrecond.sigma_max
      • EDMPrecond.sigma_data
      • EDMPrecond.model
      • EDMPrecond.__init__()
      • EDMPrecond.forward()
      • EDMPrecond.round_sigma()
  • IPSL_AID.utils module
    • EasyDict
    • FileUtils
      • FileUtils.__init__()
      • FileUtils.makedir()
      • FileUtils.makefile()
  • IPSL_AID.download_ERA5_cds module
    • parse_args()
    • main()
  • IPSL_AID.generate_all_data_ERA5 module
    • parse_args()
    • main()