Index

A

aggregator (fme.ace.InferenceConfig attribute) (fme.ace.InferenceEvaluatorConfig attribute) (fme.ace.InlineInferenceConfig attribute)

all_names (fme.ace.SingleModuleStepperConfig property) alpha (fme.ace.ParameterInitializationConfig attribute) apply() (fme.ace.CopyWeightsConfig method) (fme.ace.ParameterInitializationConfig method)

B

batch_size (fme.ace.DataLoaderConfig attribute) beta (fme.ace.ParameterInitializationConfig attribute) build() (fme.ace.ModuleSelector method) (fme.ace.SchedulerConfig method) (fme.ace.SFNO_V0_1_0 method) (fme.ace.SphericalFourierNeuralOperatorBuilder method)

builder (fme.ace.SingleModuleStepperConfig attribute)

C

checkpoint_dir (fme.ace.TrainConfig property) checkpoint_path (fme.ace.ExistingStepperConfig attribute) (fme.ace.InferenceConfig attribute) (fme.ace.InferenceEvaluatorConfig attribute) checkpoint_save_epochs (fme.ace.TrainConfig attribute) config (fme.ace.ModuleSelector attribute)

configure_logging() (fme.ace.LoggingConfig method) conserve_dry_air (fme.ace.CorrectorConfig attribute) copy_weights_after_batch (fme.ace.TrainConfig attribute) CopyWeightsConfig (class in fme.ace) corrector (fme.ace.SingleModuleStepperConfig attribute) CorrectorConfig (class in fme.ace)

D

data_path (fme.ace.XarrayDataConfig attribute) data_writer (fme.ace.InferenceConfig attribute) (fme.ace.InferenceEvaluatorConfig attribute) DataLoaderConfig (class in fme.ace)

dataset (fme.ace.DataLoaderConfig attribute) (fme.ace.ForcingDataLoaderConfig attribute) (fme.ace.InferenceDataLoaderConfig attribute) DataWriterConfig (class in fme.ace) decay (fme.ace.EMAConfig attribute)

E

ema (fme.ace.TrainConfig attribute) ema_checkpoint_save_epochs (fme.ace.TrainConfig attribute) EMAConfig (class in fme.ace) enable_automatic_mixed_precision (fme.ace.OptimizationConfig attribute) engine (fme.ace.InitialConditionConfig attribute) (fme.ace.XarrayDataConfig attribute) entity (fme.ace.LoggingConfig attribute) epochs (fme.ace.InlineInferenceConfig attribute)

exclude (fme.ace.CopyWeightsConfig attribute) (fme.ace.FrozenParameterConfig attribute) exclude_names (fme.ace.NormalizationConfig attribute) exclude_parameters (fme.ace.ParameterInitializationConfig attribute) ExistingStepperConfig (class in fme.ace) experiment_dir (fme.ace.InferenceConfig attribute) (fme.ace.InferenceEvaluatorConfig attribute) (fme.ace.TrainConfig attribute) ExplicitIndices (class in fme.ace)

F

file_pattern (fme.ace.XarrayDataConfig attribute) first (fme.ace.InferenceInitialConditionIndices attribute) fme module fme.ace module force_positive_names (fme.ace.CorrectorConfig attribute) forcing_loader (fme.ace.InferenceConfig attribute) forcing_names (fme.ace.SingleModuleStepperConfig property)

ForcingDataLoaderConfig (class in fme.ace) forward_steps_in_memory (fme.ace.InferenceConfig attribute) (fme.ace.InferenceEvaluatorConfig attribute) (fme.ace.InlineInferenceConfig attribute) from_state() (fme.ace.ModuleSelector class method) (fme.StandardNormalizer class method) FromStateNormalizer (class in fme.ace) frozen_parameters (fme.ace.ParameterInitializationConfig attribute) FrozenParameterConfig (class in fme.ace)

G

get_available_module_types() (in module fme.ace) get_base_weights() (fme.ace.ParameterInitializationConfig method) (fme.ace.SingleModuleStepperConfig method) get_device() (in module fme) get_state() (fme.ace.ModuleSelector method) (fme.StandardNormalizer method)

global_means_path (fme.ace.NormalizationConfig attribute) global_stds_path (fme.ace.NormalizationConfig attribute) gradient_magnitude() (in module fme) gradient_magnitude_percent_diff() (in module fme)

I

in_names (fme.ace.SingleModuleStepperConfig attribute) include (fme.ace.CopyWeightsConfig attribute) (fme.ace.FrozenParameterConfig attribute) infer_timestep (fme.ace.XarrayDataConfig attribute) inference (fme.ace.TrainConfig attribute) InferenceAggregatorConfig (class in fme.ace) InferenceConfig (class in fme.ace) InferenceDataLoaderConfig (class in fme.ace)

InferenceEvaluatorAggregatorConfig (class in fme.ace) InferenceEvaluatorConfig (class in fme.ace) InferenceInitialConditionIndices (class in fme.ace) initial_condition (fme.ace.InferenceConfig attribute) InitialConditionConfig (class in fme.ace) InlineInferenceConfig (class in fme.ace) interpolate (fme.ace.OceanConfig attribute) interval (fme.ace.InferenceInitialConditionIndices attribute)

K

kwargs (fme.ace.OptimizationConfig attribute) (fme.ace.SchedulerConfig attribute)

L

list (fme.ace.ExplicitIndices attribute) load_stepper() (fme.ace.InferenceConfig method) (fme.ace.InferenceEvaluatorConfig method) loader (fme.ace.InferenceEvaluatorConfig attribute) (fme.ace.InlineInferenceConfig attribute) log_extended_video (fme.ace.InferenceEvaluatorAggregatorConfig attribute) log_extended_video_netcdfs (fme.ace.DataWriterConfig attribute) log_format (fme.ace.LoggingConfig attribute) log_histograms (fme.ace.InferenceEvaluatorAggregatorConfig attribute) log_seasonal_means (fme.ace.InferenceEvaluatorAggregatorConfig attribute) log_to_file (fme.ace.LoggingConfig attribute)

log_to_screen (fme.ace.LoggingConfig attribute) log_to_wandb (fme.ace.LoggingConfig attribute) log_train_every_n_batches (fme.ace.TrainConfig attribute) log_video (fme.ace.InferenceEvaluatorAggregatorConfig attribute) log_zonal_mean_images (fme.ace.InferenceEvaluatorAggregatorConfig attribute) logging (fme.ace.InferenceConfig attribute) (fme.ace.InferenceEvaluatorConfig attribute) (fme.ace.TrainConfig attribute) LoggingConfig (class in fme.ace) loss (fme.ace.SingleModuleStepperConfig attribute) loss_normalization (fme.ace.SingleModuleStepperConfig attribute) lr (fme.ace.OptimizationConfig attribute)

M

max_epochs (fme.ace.TrainConfig attribute) means (fme.ace.NormalizationConfig attribute) mixed_layer_depth_name (fme.ace.SlabOceanConfig attribute) module fme fme.ace

ModuleSelector (class in fme.ace) moisture_budget_correction (fme.ace.CorrectorConfig attribute) monthly_reference_data (fme.ace.InferenceEvaluatorAggregatorConfig attribute)

N

n_coarsened_timesteps() (fme.ace.TimeCoarsenConfig method) n_forward_steps (fme.ace.InferenceConfig attribute) (fme.ace.InferenceEvaluatorConfig attribute) (fme.ace.InlineInferenceConfig attribute) (fme.ace.TrainConfig attribute) n_initial_conditions (fme.ace.InferenceInitialConditionIndices attribute) n_repeats (fme.ace.XarrayDataConfig attribute)

names (fme.ace.DataWriterConfig attribute) next_step_forcing_names (fme.ace.SingleModuleStepperConfig attribute) normalization (fme.ace.SingleModuleStepperConfig attribute) NormalizationConfig (class in fme.ace) normalize_names (fme.ace.SingleModuleStepperConfig property) num_data_workers (fme.ace.DataLoaderConfig attribute) (fme.ace.ForcingDataLoaderConfig attribute) (fme.ace.InferenceDataLoaderConfig attribute)

O

ocean (fme.ace.InferenceConfig attribute) (fme.ace.InferenceEvaluatorConfig attribute) (fme.ace.SingleModuleStepperConfig attribute) ocean_fraction_name (fme.ace.OceanConfig attribute)

OceanConfig (class in fme.ace) optimization (fme.ace.TrainConfig attribute) OptimizationConfig (class in fme.ace) optimizer_type (fme.ace.OptimizationConfig attribute) out_names (fme.ace.SingleModuleStepperConfig attribute)

P

pack() (fme.Packer method) Packer (class in fme) parameter_init (fme.ace.SingleModuleStepperConfig attribute) ParameterInitializationConfig (class in fme.ace)

path (fme.ace.InitialConditionConfig attribute) prediction_loader (fme.ace.InferenceEvaluatorConfig attribute) prefetch_factor (fme.ace.DataLoaderConfig attribute) prognostic_names (fme.ace.SingleModuleStepperConfig property) project (fme.ace.LoggingConfig attribute)

Q

q_flux_name (fme.ace.SlabOceanConfig attribute)

R

register() (in module fme.ace) residual_normalization (fme.ace.SingleModuleStepperConfig attribute)

rmse_of_time_mean() (in module fme) root_mean_squared_error() (in module fme)

S

save_checkpoint (fme.ace.TrainConfig attribute) save_histogram_files (fme.ace.DataWriterConfig attribute) save_monthly_files (fme.ace.DataWriterConfig attribute) save_prediction_files (fme.ace.DataWriterConfig attribute) scheduler (fme.ace.OptimizationConfig attribute) SchedulerConfig (class in fme.ace) segment_epochs (fme.ace.TrainConfig attribute) SFNO_V0_1_0 (class in fme.ace) SingleModuleStepperConfig (class in fme.ace) slab (fme.ace.OceanConfig attribute) SlabOceanConfig (class in fme.ace) Slice (class in fme.ace) spatial_dimensions (fme.ace.XarrayDataConfig attribute) spherical_area_weights() (in module fme) SphericalFourierNeuralOperatorBuilder (class in fme.ace)

StandardNormalizer (class in fme) start (fme.ace.Slice attribute) start_indices (fme.ace.InferenceDataLoaderConfig attribute) (fme.ace.InitialConditionConfig attribute) start_time (fme.ace.TimeSlice attribute) state (fme.ace.FromStateNormalizer attribute) stds (fme.ace.NormalizationConfig attribute) step (fme.ace.Slice attribute) (fme.ace.TimeSlice attribute) stepper (fme.ace.TrainConfig attribute) stop (fme.ace.Slice attribute) stop_time (fme.ace.TimeSlice attribute) strict_ensemble (fme.ace.DataLoaderConfig attribute) subset (fme.ace.XarrayDataConfig attribute) surface_temperature_name (fme.ace.OceanConfig attribute)

T

time_and_global_mean_bias() (in module fme) time_coarsen (fme.ace.DataWriterConfig attribute) time_mean_reference_data (fme.ace.InferenceAggregatorConfig attribute) (fme.ace.InferenceEvaluatorAggregatorConfig attribute) TimeCoarsenConfig (class in fme.ace) times (fme.ace.TimestampList attribute)

TimeSlice (class in fme.ace) timestamp_format (fme.ace.TimestampList attribute) TimestampList (class in fme.ace) train_loader (fme.ace.TrainConfig attribute) TrainConfig (class in fme.ace) type (fme.ace.ModuleSelector attribute) (fme.ace.SchedulerConfig attribute)

V

validate_using_ema (fme.ace.TrainConfig attribute)

validation_loader (fme.ace.TrainConfig attribute)

W

weight_path (fme.ace.ParameterInitializationConfig attribute) weighted_mean() (in module fme)

weighted_mean_bias() (in module fme) weighted_mean_gradient_magnitude() (in module fme) WeightedMappingLossConfig (class in fme.ace)

X

XarrayDataConfig (class in fme.ace)

Z

zero_global_mean_moisture_advection (fme.ace.CorrectorConfig attribute)