Introduction
Overview
Installing
User Guide
User Journey
Basic Set-Up
Configuration Basics
Basic Training Configuration
Multiple Dataset Support
Training
Evaluation
Tasks
Training Methods
Models
Tracking
Benchmarking
Parallelisation
ERA5 training data
Performance Optimisation
Checkpoint Pipeline
Checkpoint Pipeline (Developer Guide)
Checkpoint Pipeline Configuration
Checkpoint Pipeline Troubleshooting
API Reference
Data
Diagnostics
Losses
Optimization
Schemas
Strategy
Tasks
Model Training
Contributing
General guidelines
Configuration guidelines
Integration tests and use cases
FAQ/Troubleshooting
Troubleshooting
Anemoi Training
Index
Index
A
|
B
|
C
|
D
|
E
|
F
|
G
|
H
|
I
|
K
|
L
|
M
|
N
|
O
|
P
|
R
|
S
|
T
|
U
|
V
|
W
|
Y
A
accelerator (anemoi.training.schemas.system.HardwareSchema attribute)
accum_grad_batches (anemoi.training.schemas.training.BaseTrainingSchema attribute)
add_empty_scalers_to_inner() (anemoi.training.schemas.training.MultiScaleLossSchema class method)
(anemoi.training.schemas.training.TimeAggregateLossWrapperSchema class method)
add_scaler() (anemoi.training.losses.base.BaseLoss method)
(anemoi.training.losses.base.BaseLossWrapper method)
(anemoi.training.losses.combined.CombinedLoss method)
AdEMAMix (class in anemoi.training.optimizers.AdEMAMix)
algorithm (anemoi.training.schemas.training.GradientClip attribute)
allow_variable_subset (anemoi.training.schemas.training.BaseTrainingSchema attribute)
alpha (anemoi.training.schemas.training.CRPSSchema attribute)
anemoi.graphs.schemas.base_graph
module
anemoi.models.schemas.models
module
anemoi.training.data.data_reader
module
anemoi.training.data.loading
module
anemoi.training.data.multidataset
module
anemoi.training.diagnostics.evaluation.plotting.protocols
module
anemoi.training.losses.base
module
anemoi.training.losses.combined
module
anemoi.training.losses.utils
module
anemoi.training.optimizers.AdEMAMix
module
anemoi.training.schemas.dataloader
module
anemoi.training.schemas.system
module
anemoi.training.schemas.training
module
attributes (anemoi.graphs.schemas.base_graph.EdgeSchema attribute)
(anemoi.graphs.schemas.base_graph.NodeSchema attribute)
autocast (anemoi.models.schemas.models.NoiseConditioningSchema attribute)
B
backend (anemoi.training.schemas.training.CRPSSchema attribute)
BaseAnemoiReader (class in anemoi.training.data.data_reader)
BaseDDPStrategySchema (class in anemoi.training.schemas.training)
BaseGraphSchema (class in anemoi.graphs.schemas.base_graph)
BaseLoss (class in anemoi.training.losses.base)
BaseLossSchema (class in anemoi.training.schemas.training)
BaseLossWrapper (class in anemoi.training.losses.base)
BaseModelSchema (class in anemoi.models.schemas.models)
BaseTrainingSchema (class in anemoi.training.schemas.training)
batch_size (anemoi.training.schemas.dataloader.DataLoaderSchema attribute)
BatchOutputPlotFn (class in anemoi.training.diagnostics.evaluation.plotting.protocols)
Boolean1DSchema (class in anemoi.models.schemas.models)
boundary_mask (anemoi.training.data.data_reader.BaseAnemoiReader property)
bounding (anemoi.models.schemas.models.BaseModelSchema attribute)
broadcast_buffers (anemoi.training.schemas.training.BaseDDPStrategySchema attribute)
build_batch_sampler() (anemoi.training.data.loading.DataLoadingPolicy method)
(anemoi.training.data.loading.SynchronizedLoadingPolicy method)
build_dataset() (anemoi.training.data.loading.DataLoadingPolicy method)
(anemoi.training.data.loading.SynchronizedLoadingPolicy method)
C
calculate_difference() (anemoi.training.losses.base.FunctionalLoss method)
cast_encoder_decoder_keys_to_str() (anemoi.models.schemas.models.BaseModelSchema class method)
check_length_of_weights_and_losses() (anemoi.training.schemas.training.CombinedLossSchema method)
check_loss_tree_variable_units() (in module anemoi.training.losses.utils)
check_static_graph_without_accumulation() (anemoi.training.schemas.training.BaseTrainingSchema method)
check_variables_compatibility (anemoi.training.schemas.training.BaseLossSchema attribute)
(anemoi.training.schemas.training.BaseTrainingSchema attribute)
checkpoint (anemoi.training.schemas.training.BaseTrainingSchema attribute)
CheckpointPipelineSchema (class in anemoi.training.schemas.training)
checkpoints (anemoi.training.schemas.system.OutputSchema attribute)
CheckpointsSchema (class in anemoi.training.schemas.system)
CheckVariablesCompatibilitySchema (class in anemoi.training.schemas.training)
coefficient_magnitude (anemoi.training.schemas.training.SpectralLossSchema attribute)
CombinedLoss (class in anemoi.training.losses.combined)
CombinedLossSchema (class in anemoi.training.schemas.training)
compile (anemoi.models.schemas.models.BaseModelSchema attribute)
compute_anchors() (anemoi.training.data.data_reader.BaseAnemoiReader method)
condition_on_residual (anemoi.models.schemas.models.EnsModelSchema attribute)
(anemoi.models.schemas.models.TransportTendModelSchema attribute)
convert_ (anemoi.models.schemas.models.Model attribute)
create_dataset() (in module anemoi.training.data.data_reader)
CRPSSchema (class in anemoi.training.schemas.training)
cutout_mask (anemoi.training.data.data_reader.BaseAnemoiReader property)
D
data (anemoi.training.data.multidataset.MultiDataset property)
(anemoi.training.schemas.training.TargetForcing attribute)
DataLoaderSchema (class in anemoi.training.schemas.dataloader)
DataLoadingPolicy (class in anemoi.training.data.loading)
DataLoadingPolicySchema (class in anemoi.training.schemas.dataloader)
dataset (anemoi.training.schemas.dataloader.DatasetConfigSchema attribute)
(anemoi.training.schemas.system.InputSchema attribute)
dataset_config (anemoi.training.schemas.dataloader.NativeDatasetSchema attribute)
dataset_fusing_strategy (anemoi.models.schemas.models.EncodersSchema attribute)
DatasetConfigSchema (class in anemoi.training.schemas.dataloader)
dates (anemoi.training.data.data_reader.BaseAnemoiReader property)
DDPEnsGroupStrategy (class in anemoi.training.distributed.strategy)
DDPEnsGroupStrategyStrategySchema (class in anemoi.training.schemas.training)
DDPGroupStrategy (class in anemoi.training.distributed.strategy)
decay_factor (anemoi.training.schemas.training.LeadTimeDecayScalerSchema attribute)
decay_type (anemoi.training.schemas.training.LeadTimeDecayScalerSchema attribute)
decoders (anemoi.models.schemas.models.BaseModelSchema attribute)
DecodersSchema (class in anemoi.models.schemas.models)
default (anemoi.training.schemas.training.LossScalingSchema attribute)
default_num_channels_in_hierarchical_mapper() (anemoi.models.schemas.models.HierarchicalModelSchema class method)
default_sampling (anemoi.training.data.data_reader.BaseAnemoiReader attribute)
DefinedModels (class in anemoi.models.schemas.models)
delta (anemoi.training.schemas.training.HuberLossSchema attribute)
deterministic (anemoi.training.schemas.training.BaseTrainingSchema attribute)
device (anemoi.training.schemas.training.SingleDeviceStrategySchema attribute)
downscale_mapper (anemoi.models.schemas.models.HierarchicalModelSchema attribute)
drop (anemoi.training.schemas.dataloader.DatasetConfigSchema attribute)
E
edge_builders (anemoi.graphs.schemas.base_graph.EdgeSchema attribute)
edge_weight_attribute (anemoi.models.schemas.models.NoiseConditioningSchema attribute)
edges (anemoi.graphs.schemas.base_graph.BaseGraphSchema attribute)
EdgeSchema (class in anemoi.graphs.schemas.base_graph)
enable_hierarchical_level_processing (anemoi.models.schemas.models.HierarchicalModelSchema attribute)
encoders (anemoi.models.schemas.models.BaseModelSchema attribute)
EncodersSchema (class in anemoi.models.schemas.models)
end (anemoi.training.schemas.dataloader.NativeDatasetSchema attribute)
EnergyScoreLossSchema (class in anemoi.training.schemas.training)
ensemble_size_per_device (anemoi.training.schemas.training.BaseTrainingSchema attribute)
EnsembleScoreLossSchema (class in anemoi.training.schemas.training)
EnsembleTrainingMethodSchema (class in anemoi.training.schemas.training)
EnsembleTrainingSchema (class in anemoi.training.schemas.training)
EnsModelSchema (class in anemoi.models.schemas.models)
every_n_epochs (anemoi.training.schemas.system.CheckpointsSchema attribute)
every_n_minutes (anemoi.training.schemas.system.CheckpointsSchema attribute)
every_n_train_steps (anemoi.training.schemas.system.CheckpointsSchema attribute)
ExplicitTimes (class in anemoi.training.schemas.training)
F
fake_dataloading (anemoi.training.schemas.dataloader.DataLoaderSchema attribute)
find_unused_parameters (anemoi.training.schemas.training.BaseDDPStrategySchema attribute)
forward() (anemoi.training.losses.base.BaseLoss method)
(anemoi.training.losses.base.FunctionalLoss method)
(anemoi.training.losses.combined.CombinedLoss method)
frequency (anemoi.training.data.data_reader.BaseAnemoiReader property)
(anemoi.training.data.data_reader.TrajectoryDataset property)
(anemoi.training.data.multidataset.MultiDataset property)
(anemoi.training.schemas.dataloader.DatasetConfigSchema attribute)
Frequency (class in anemoi.training.schemas.dataloader)
FunctionalLoss (class in anemoi.training.losses.base)
G
GeneralVariableLossScalerSchema (class in anemoi.training.schemas.training)
GenericSchema (class in anemoi.training.schemas.training)
get_sample() (anemoi.training.data.data_reader.BaseAnemoiReader method)
(anemoi.training.data.data_reader.TrajectoryDataset method)
gradient_clip (anemoi.training.schemas.training.BaseTrainingSchema attribute)
GradientClip (class in anemoi.training.schemas.training)
graph (anemoi.training.schemas.system.InputSchema attribute)
GraphEdgeCRPSLossSchema (class in anemoi.training.schemas.training)
GraphEdgeEnergyScoreLossSchema (class in anemoi.training.schemas.training)
GraphEnergyScoreLossSchema (class in anemoi.training.schemas.training)
GraphLossMatrixSchema (class in anemoi.training.schemas.training)
GraphNodeAttributeScalerSchema (class in anemoi.training.schemas.training)
GraphPlotFn (class in anemoi.training.diagnostics.evaluation.plotting.protocols)
GraphScoreGraphSchema (class in anemoi.training.schemas.training)
GraphScoreLossSchema (class in anemoi.training.schemas.training)
GraphVariogramScoreLossSchema (class in anemoi.training.schemas.training)
grid_size (anemoi.training.data.data_reader.BaseAnemoiReader property)
group (anemoi.training.schemas.training.VariableLevelScalerSchema attribute)
H
hardware (anemoi.training.schemas.system.SystemSchema attribute)
HardwareSchema (class in anemoi.training.schemas.system)
has_scaler_for_dim() (anemoi.training.losses.base.BaseLoss method)
(anemoi.training.losses.base.BaseLossWrapper method)
(anemoi.training.losses.combined.CombinedLoss method)
hidden_nodes_name (anemoi.models.schemas.models.Model attribute)
HierarchicalModelSchema (class in anemoi.models.schemas.models)
HuberLossSchema (class in anemoi.training.schemas.training)
I
ignore_nans (anemoi.training.schemas.training.BaseLossSchema attribute)
ignore_processing_period (anemoi.training.schemas.training.CheckVariablesCompatibilitySchema attribute)
ignore_time_processing (anemoi.training.schemas.training.CheckVariablesCompatibilitySchema attribute)
ignore_type_of_level (anemoi.training.schemas.training.CheckVariablesCompatibilitySchema attribute)
ignore_units (anemoi.training.schemas.training.CheckVariablesCompatibilitySchema attribute)
ImplementedLossesUsingBaseLossSchema (class in anemoi.training.schemas.training)
ImplementedStrategiesUsingBaseDDPStrategySchema (class in anemoi.training.schemas.training)
inference_defaults (anemoi.models.schemas.models.TransportConfig attribute)
input (anemoi.training.schemas.system.SystemSchema attribute)
(anemoi.training.schemas.training.ExplicitTimes attribute)
InputSchema (class in anemoi.training.schemas.system)
interval (anemoi.training.schemas.training.PLSchedulerSchema attribute)
inverse (anemoi.training.schemas.training.LeadTimeDecayScalerSchema attribute)
invert (anemoi.training.schemas.training.VariableMaskingScalerSchema attribute)
iter_leaf_losses() (anemoi.training.losses.base.BaseLoss method)
(anemoi.training.losses.base.BaseLossWrapper method)
(anemoi.training.losses.combined.CombinedLoss method)
K
keep_batch_sharded (anemoi.models.schemas.models.BaseModelSchema attribute)
kind (anemoi.models.schemas.models.TransportSourceConfig attribute)
L
latent_aggregator (anemoi.models.schemas.models.BaseModelSchema attribute)
latent_skip (anemoi.models.schemas.models.BaseModelSchema attribute)
(anemoi.models.schemas.models.Model attribute)
layer_kernels (anemoi.models.schemas.models.NoiseConditioningSchema attribute)
(anemoi.models.schemas.models.NoiseInjectorSchema attribute)
LeadTimeDecayScalerSchema (class in anemoi.training.schemas.training)
level_process_num_layers (anemoi.models.schemas.models.HierarchicalModelSchema attribute)
limit_batches (anemoi.training.schemas.dataloader.DataLoaderSchema attribute)
LoaderSet (class in anemoi.training.schemas.dataloader)
loading (anemoi.training.schemas.training.CheckpointPipelineSchema attribute)
loading_policy (anemoi.training.schemas.dataloader.DataLoaderSchema attribute)
logs (anemoi.training.schemas.system.OutputSchema attribute)
Logs (class in anemoi.training.schemas.system)
loss_fn (anemoi.training.schemas.training.TimeAggregateLossWrapperSchema attribute)
loss_gradient_scaling (anemoi.training.schemas.training.BaseTrainingSchema attribute)
loss_matrices_path (anemoi.training.schemas.system.InputSchema attribute)
loss_weights (anemoi.training.schemas.training.CombinedLossSchema attribute)
losses (anemoi.training.schemas.training.CombinedLossSchema attribute)
LossFactoryContextKey (class in anemoi.training.losses.base)
LossPlotFn (class in anemoi.training.diagnostics.evaluation.plotting.protocols)
LossScalingSchema (class in anemoi.training.schemas.training)
lr (anemoi.training.schemas.training.OptimizationSchema attribute)
lr_scheduler (anemoi.training.schemas.training.OptimizationSchema attribute)
LRSchedulerSchema (class in anemoi.training.schemas.training)
M
mask_nans() (anemoi.training.losses.base.BaseLoss method)
max_epochs (anemoi.training.schemas.training.BaseTrainingSchema attribute)
max_lead_time (anemoi.training.schemas.training.LeadTimeDecayScalerSchema attribute)
max_steps (anemoi.training.schemas.training.BaseTrainingSchema attribute)
metadata (anemoi.training.data.data_reader.BaseAnemoiReader property)
(anemoi.training.data.multidataset.MultiDataset property)
method (anemoi.training.schemas.training.EnsembleTrainingSchema attribute)
(anemoi.training.schemas.training.SingleTrainingSchema attribute)
(anemoi.training.schemas.training.TransportTrainingSchema attribute)
metrics (anemoi.training.schemas.training.BaseTrainingSchema attribute)
missing (anemoi.training.data.data_reader.BaseAnemoiReader property)
missing_positions() (anemoi.training.data.data_reader.BaseAnemoiReader method)
(anemoi.training.data.data_reader.TrajectoryDataset method)
missing_sequences (anemoi.training.data.data_reader.BaseAnemoiReader property)
(anemoi.training.data.data_reader.TrajectoryDataset property)
mlflow (anemoi.training.schemas.system.Logs attribute)
model (anemoi.models.schemas.models.BaseModelSchema attribute)
(anemoi.models.schemas.models.TransportModelSchema attribute)
Model (class in anemoi.models.schemas.models)
model_config (anemoi.graphs.schemas.base_graph.BaseGraphSchema attribute)
(anemoi.graphs.schemas.base_graph.EdgeSchema attribute)
(anemoi.graphs.schemas.base_graph.NodeSchema attribute)
(anemoi.models.schemas.models.BaseModelSchema attribute)
(anemoi.models.schemas.models.Boolean1DSchema attribute)
(anemoi.models.schemas.models.DecodersSchema attribute)
(anemoi.models.schemas.models.EncodersSchema attribute)
(anemoi.models.schemas.models.EnsModelSchema attribute)
(anemoi.models.schemas.models.HierarchicalModelSchema attribute)
(anemoi.models.schemas.models.Model attribute)
(anemoi.models.schemas.models.NoiseConditioningSchema attribute)
(anemoi.models.schemas.models.NoiseInjectorSchema attribute)
(anemoi.models.schemas.models.NoOpNoiseInjectorSchema attribute)
(anemoi.models.schemas.models.NoOutputMaskSchema attribute)
(anemoi.models.schemas.models.SparseProjectorSchema attribute)
(anemoi.models.schemas.models.TransportConfig attribute)
(anemoi.models.schemas.models.TransportModel attribute)
(anemoi.models.schemas.models.TransportModelSchema attribute)
(anemoi.models.schemas.models.TransportSourceConfig attribute)
(anemoi.models.schemas.models.TransportTendModelSchema attribute)
(anemoi.training.schemas.dataloader.DataLoaderSchema attribute)
(anemoi.training.schemas.dataloader.DataLoadingPolicySchema attribute)
(anemoi.training.schemas.dataloader.DatasetConfigSchema attribute)
(anemoi.training.schemas.dataloader.Frequency attribute)
(anemoi.training.schemas.dataloader.LoaderSet attribute)
(anemoi.training.schemas.dataloader.NativeDatasetSchema attribute)
(anemoi.training.schemas.dataloader.TrajectoryDatasetSchema attribute)
(anemoi.training.schemas.dataloader.TrajectorySamplingSchema attribute)
(anemoi.training.schemas.dataloader.TrajectorySchema attribute)
(anemoi.training.schemas.system.CheckpointsSchema attribute)
(anemoi.training.schemas.system.HardwareSchema attribute)
(anemoi.training.schemas.system.InputSchema attribute)
(anemoi.training.schemas.system.Logs attribute)
(anemoi.training.schemas.system.OutputSchema attribute)
(anemoi.training.schemas.system.SystemSchema attribute)
(anemoi.training.schemas.training.BaseDDPStrategySchema attribute)
(anemoi.training.schemas.training.BaseLossSchema attribute)
(anemoi.training.schemas.training.BaseTrainingSchema attribute)
(anemoi.training.schemas.training.CheckpointPipelineSchema attribute)
(anemoi.training.schemas.training.CheckVariablesCompatibilitySchema attribute)
(anemoi.training.schemas.training.CombinedLossSchema attribute)
(anemoi.training.schemas.training.CRPSSchema attribute)
(anemoi.training.schemas.training.DDPEnsGroupStrategyStrategySchema attribute)
(anemoi.training.schemas.training.EnergyScoreLossSchema attribute)
(anemoi.training.schemas.training.EnsembleScoreLossSchema attribute)
(anemoi.training.schemas.training.EnsembleTrainingMethodSchema attribute)
(anemoi.training.schemas.training.EnsembleTrainingSchema attribute)
(anemoi.training.schemas.training.ExplicitTimes attribute)
(anemoi.training.schemas.training.GeneralVariableLossScalerSchema attribute)
(anemoi.training.schemas.training.GenericSchema attribute)
(anemoi.training.schemas.training.GradientClip attribute)
(anemoi.training.schemas.training.GraphEdgeCRPSLossSchema attribute)
(anemoi.training.schemas.training.GraphEdgeEnergyScoreLossSchema attribute)
(anemoi.training.schemas.training.GraphEnergyScoreLossSchema attribute)
(anemoi.training.schemas.training.GraphLossMatrixSchema attribute)
(anemoi.training.schemas.training.GraphNodeAttributeScalerSchema attribute)
(anemoi.training.schemas.training.GraphScoreGraphSchema attribute)
(anemoi.training.schemas.training.GraphScoreLossSchema attribute)
(anemoi.training.schemas.training.GraphVariogramScoreLossSchema attribute)
(anemoi.training.schemas.training.HuberLossSchema attribute)
(anemoi.training.schemas.training.LeadTimeDecayScalerSchema attribute)
(anemoi.training.schemas.training.LossScalingSchema attribute)
(anemoi.training.schemas.training.LRSchedulerSchema attribute)
(anemoi.training.schemas.training.MultiscaleConfigDiskSchema attribute)
(anemoi.training.schemas.training.MultiscaleConfigOnTheFlySchema attribute)
(anemoi.training.schemas.training.MultiScaleLossSchema attribute)
(anemoi.training.schemas.training.NaNMaskScalerSchema attribute)
(anemoi.training.schemas.training.OptimizationSchema attribute)
(anemoi.training.schemas.training.OptimizerSchema attribute)
(anemoi.training.schemas.training.PLSchedulerSchema attribute)
(anemoi.training.schemas.training.ReweightedGraphNodeAttributeScalerSchema attribute)
(anemoi.training.schemas.training.SingleDeviceStrategySchema attribute)
(anemoi.training.schemas.training.SingleTrainingMethodSchema attribute)
(anemoi.training.schemas.training.SingleTrainingSchema attribute)
(anemoi.training.schemas.training.SpectralDimensionScalerSchema attribute)
(anemoi.training.schemas.training.SpectralLossSchema attribute)
(anemoi.training.schemas.training.SpectralProjectionConfigSchema attribute)
(anemoi.training.schemas.training.TargetForcing attribute)
(anemoi.training.schemas.training.TendencyScalerSchema attribute)
(anemoi.training.schemas.training.TimeAggregateLossWrapperSchema attribute)
(anemoi.training.schemas.training.TimeStepScalerSchema attribute)
(anemoi.training.schemas.training.TransportTrainingConfigSchema attribute)
(anemoi.training.schemas.training.TransportTrainingMethodSchema attribute)
(anemoi.training.schemas.training.TransportTrainingSchema attribute)
(anemoi.training.schemas.training.UniformTimeStepScalerSchema attribute)
(anemoi.training.schemas.training.UpdateDsStatsOnCkptLoadSchema attribute)
(anemoi.training.schemas.training.VariableLevelScalerSchema attribute)
(anemoi.training.schemas.training.VariableMaskingScalerSchema attribute)
(anemoi.training.schemas.training.WeightAveragingSchema attribute)
modifiers (anemoi.training.schemas.training.CheckpointPipelineSchema attribute)
module
anemoi.graphs.schemas.base_graph
anemoi.models.schemas.models
anemoi.training.data.data_reader
anemoi.training.data.loading
anemoi.training.data.multidataset
anemoi.training.diagnostics.evaluation.plotting.protocols
anemoi.training.losses.base
anemoi.training.losses.combined
anemoi.training.losses.utils
anemoi.training.optimizers.AdEMAMix
anemoi.training.schemas.dataloader
anemoi.training.schemas.system
anemoi.training.schemas.training
MultiDataset (class in anemoi.training.data.multidataset)
multiprocessing_context (anemoi.training.schemas.dataloader.DataLoaderSchema attribute)
MultiscaleConfigDiskSchema (class in anemoi.training.schemas.training)
MultiscaleConfigOnTheFlySchema (class in anemoi.training.schemas.training)
MultiScaleLossSchema (class in anemoi.training.schemas.training)
N
n_spectral_modes (anemoi.training.schemas.training.SpectralDimensionScalerSchema attribute)
name (anemoi.training.losses.base.BaseLoss property)
name_to_index (anemoi.training.data.data_reader.BaseAnemoiReader property)
(anemoi.training.data.multidataset.MultiDataset property)
NaNMaskScalerSchema (class in anemoi.training.schemas.training)
NativeDatasetSchema (class in anemoi.training.schemas.dataloader)
NativeGridDataset (class in anemoi.training.data.data_reader)
needs_shard_layout_info (anemoi.training.losses.base.BaseLoss property)
(anemoi.training.losses.base.BaseLossWrapper property)
(anemoi.training.losses.combined.CombinedLoss property)
no_autocast (anemoi.training.schemas.training.CRPSSchema attribute)
node_builder (anemoi.graphs.schemas.base_graph.NodeSchema attribute)
node_trainable_parameters (anemoi.models.schemas.models.BaseModelSchema attribute)
nodes (anemoi.graphs.schemas.base_graph.BaseGraphSchema attribute)
nodes_attribute_name (anemoi.training.schemas.training.GraphNodeAttributeScalerSchema attribute)
(anemoi.training.schemas.training.ReweightedGraphNodeAttributeScalerSchema attribute)
NodeSchema (class in anemoi.graphs.schemas.base_graph)
noise_channels (anemoi.models.schemas.models.TransportConfig attribute)
noise_channels_dim (anemoi.models.schemas.models.NoiseConditioningSchema attribute)
(anemoi.models.schemas.models.NoiseInjectorSchema attribute)
noise_cond_dim (anemoi.models.schemas.models.TransportConfig attribute)
noise_edges_name (anemoi.models.schemas.models.NoiseConditioningSchema attribute)
noise_embedder (anemoi.models.schemas.models.TransportConfig attribute)
noise_injector (anemoi.models.schemas.models.EnsModelSchema attribute)
noise_matrix (anemoi.models.schemas.models.NoiseConditioningSchema attribute)
noise_mlp_hidden_dim (anemoi.models.schemas.models.NoiseConditioningSchema attribute)
(anemoi.models.schemas.models.NoiseInjectorSchema attribute)
noise_scale (anemoi.models.schemas.models.TransportSourceConfig attribute)
noise_std (anemoi.models.schemas.models.NoiseConditioningSchema attribute)
(anemoi.models.schemas.models.NoiseInjectorSchema attribute)
NoiseConditioningSchema (class in anemoi.models.schemas.models)
NoiseInjectorSchema (class in anemoi.models.schemas.models)
NoOpNoiseInjectorSchema (class in anemoi.models.schemas.models)
NoOutputMaskSchema (class in anemoi.models.schemas.models)
norm (anemoi.training.schemas.training.GraphNodeAttributeScalerSchema attribute)
(anemoi.training.schemas.training.LeadTimeDecayScalerSchema attribute)
(anemoi.training.schemas.training.ReweightedGraphNodeAttributeScalerSchema attribute)
(anemoi.training.schemas.training.SpectralDimensionScalerSchema attribute)
(anemoi.training.schemas.training.TimeStepScalerSchema attribute)
num_chunks (anemoi.models.schemas.models.SparseProjectorSchema attribute)
num_gpus_per_ensemble (anemoi.training.schemas.system.HardwareSchema attribute)
(anemoi.training.schemas.training.DDPEnsGroupStrategyStrategySchema attribute)
num_gpus_per_model (anemoi.training.schemas.system.HardwareSchema attribute)
(anemoi.training.schemas.training.BaseDDPStrategySchema attribute)
num_gpus_per_node (anemoi.training.schemas.system.HardwareSchema attribute)
num_nodes (anemoi.training.schemas.system.HardwareSchema attribute)
num_sanity_val_steps (anemoi.training.schemas.training.BaseTrainingSchema attribute)
num_sequences (anemoi.training.data.data_reader.BaseAnemoiReader property)
(anemoi.training.data.data_reader.TrajectoryDataset property)
num_workers (anemoi.training.schemas.dataloader.DataLoaderSchema attribute)
O
objective (anemoi.models.schemas.models.TransportConfig attribute)
(anemoi.training.schemas.training.TransportTrainingConfigSchema attribute)
optimization (anemoi.training.schemas.training.BaseTrainingSchema attribute)
OptimizationSchema (class in anemoi.training.schemas.training)
optimizer (anemoi.training.schemas.training.OptimizationSchema attribute)
OptimizerSchema (class in anemoi.training.schemas.training)
output (anemoi.training.schemas.system.SystemSchema attribute)
output_lead_times (anemoi.training.schemas.training.LeadTimeDecayScalerSchema attribute)
output_mask (anemoi.models.schemas.models.BaseModelSchema attribute)
OutputSchema (class in anemoi.training.schemas.system)
overwrite (anemoi.graphs.schemas.base_graph.BaseGraphSchema attribute)
P
persistent_workers (anemoi.training.schemas.dataloader.DataLoaderSchema attribute)
pin_memory (anemoi.training.schemas.dataloader.DataLoaderSchema attribute)
pl (anemoi.training.schemas.training.LossScalingSchema attribute)
pl_lr_scheduler (anemoi.training.schemas.training.OptimizationSchema attribute)
plots (anemoi.training.schemas.system.OutputSchema attribute)
PLSchedulerSchema (class in anemoi.training.schemas.training)
precision (anemoi.training.schemas.training.BaseTrainingSchema attribute)
prediction_mode (anemoi.training.schemas.training.TransportTrainingConfigSchema attribute)
preferred_blas_backend (anemoi.training.schemas.training.BaseTrainingSchema attribute)
prefetch_factor (anemoi.training.schemas.dataloader.DataLoaderSchema attribute)
print_variable_scaling() (in module anemoi.training.losses.utils)
processor (anemoi.models.schemas.models.BaseModelSchema attribute)
profiler (anemoi.training.schemas.system.OutputSchema attribute)
projection_config (anemoi.training.schemas.training.SpectralLossSchema attribute)
propagate_scalers_to_children() (anemoi.training.schemas.training.CombinedLossSchema class method)
R
read_group_size (anemoi.training.schemas.dataloader.DataLoaderSchema attribute)
(anemoi.training.schemas.training.BaseDDPStrategySchema attribute)
recompile_limit (anemoi.models.schemas.models.BaseModelSchema attribute)
reduce() (anemoi.training.losses.base.BaseLoss method)
reduce_to_last_dim() (in module anemoi.training.losses.utils)
register_parameter_hooks() (anemoi.training.distributed.strategy.DDPEnsGroupStrategy method)
residual (anemoi.models.schemas.models.BaseModelSchema attribute)
resolution (anemoi.training.data.data_reader.BaseAnemoiReader property)
(anemoi.training.data.multidataset.MultiDataset property)
ReweightedGraphNodeAttributeScalerSchema (class in anemoi.training.schemas.training)
rho (anemoi.models.schemas.models.TransportConfig attribute)
root (anemoi.training.schemas.system.CheckpointsSchema attribute)
(anemoi.training.schemas.system.OutputSchema attribute)
row_normalize_noise_matrix (anemoi.models.schemas.models.NoiseConditioningSchema attribute)
S
sample_comm_group_size (anemoi.training.distributed.strategy.DDPEnsGroupStrategy property)
sampling (anemoi.training.schemas.dataloader.TrajectorySchema attribute)
scale (anemoi.models.schemas.models.TransportSourceConfig attribute)
scale() (anemoi.training.losses.base.BaseLoss method)
scalers (anemoi.training.schemas.training.BaseLossSchema attribute)
(anemoi.training.schemas.training.BaseTrainingSchema attribute)
(anemoi.training.schemas.training.CombinedLossSchema attribute)
(anemoi.training.schemas.training.MultiscaleConfigDiskSchema attribute)
(anemoi.training.schemas.training.TimeAggregateLossWrapperSchema attribute)
scaling_mask_attribute_name (anemoi.training.schemas.training.ReweightedGraphNodeAttributeScalerSchema attribute)
seed (anemoi.training.schemas.training.BaseTrainingSchema attribute)
select (anemoi.training.schemas.dataloader.DatasetConfigSchema attribute)
sequence_length() (anemoi.training.data.data_reader.BaseAnemoiReader method)
(anemoi.training.data.data_reader.TrajectoryDataset method)
set_reader_group_info() (anemoi.training.data.multidataset.MultiDataset method)
set_relative_date_indices() (anemoi.training.data.multidataset.MultiDataset method)
sfc (anemoi.training.schemas.training.LossScalingSchema attribute)
si_alpha_schedule (anemoi.models.schemas.models.TransportConfig attribute)
si_beta_schedule (anemoi.models.schemas.models.TransportConfig attribute)
si_noise_scale (anemoi.models.schemas.models.TransportConfig attribute)
si_sigma_schedule (anemoi.models.schemas.models.TransportConfig attribute)
sigma_data (anemoi.models.schemas.models.TransportConfig attribute)
sigma_max (anemoi.models.schemas.models.TransportConfig attribute)
sigma_min (anemoi.models.schemas.models.TransportConfig attribute)
SingleDeviceStrategySchema (class in anemoi.training.schemas.training)
SingleTrainingMethodSchema (class in anemoi.training.schemas.training)
SingleTrainingSchema (class in anemoi.training.schemas.training)
slope (anemoi.training.schemas.training.VariableLevelScalerSchema attribute)
source (anemoi.models.schemas.models.TransportConfig attribute)
(anemoi.training.schemas.training.CheckpointPipelineSchema attribute)
source_datasets (anemoi.models.schemas.models.EncodersSchema attribute)
source_name (anemoi.graphs.schemas.base_graph.EdgeSchema attribute)
sparse_projector (anemoi.models.schemas.models.BaseModelSchema attribute)
SparseProjectorSchema (class in anemoi.models.schemas.models)
spectral_dims (anemoi.training.schemas.training.SpectralDimensionScalerSchema attribute)
SpectralDimensionScalerSchema (class in anemoi.training.schemas.training)
SpectralLossSchema (class in anemoi.training.schemas.training)
SpectralLossSchema.Config (class in anemoi.training.schemas.training)
SpectralProjectionConfigSchema (class in anemoi.training.schemas.training)
start (anemoi.training.schemas.dataloader.NativeDatasetSchema attribute)
states (anemoi.training.schemas.training.UpdateDsStatsOnCkptLoadSchema attribute)
static_graph (anemoi.training.schemas.training.BaseDDPStrategySchema attribute)
statistics (anemoi.training.data.data_reader.BaseAnemoiReader property)
(anemoi.training.data.multidataset.MultiDataset property)
(anemoi.training.schemas.dataloader.DatasetConfigSchema attribute)
statistics_tendencies() (anemoi.training.data.data_reader.BaseAnemoiReader method)
step() (anemoi.training.optimizers.AdEMAMix.AdEMAMix method)
step_end (anemoi.training.schemas.dataloader.DatasetConfigSchema attribute)
step_frequency (anemoi.training.schemas.dataloader.DatasetConfigSchema attribute)
step_start (anemoi.training.schemas.dataloader.DatasetConfigSchema attribute)
strategy (anemoi.training.schemas.training.BaseTrainingSchema attribute)
stride (anemoi.training.schemas.dataloader.TrajectorySamplingSchema attribute)
subgrid (anemoi.training.schemas.training.SpectralLossSchema attribute)
supporting_arrays (anemoi.training.data.data_reader.BaseAnemoiReader property)
(anemoi.training.data.multidataset.MultiDataset property)
SynchronizedLoadingPolicy (class in anemoi.training.data.loading)
SystemSchema (class in anemoi.training.schemas.system)
T
target (anemoi.training.schemas.training.ExplicitTimes attribute)
target_ (anemoi.models.schemas.models.Model attribute)
(anemoi.models.schemas.models.NoiseConditioningSchema attribute)
(anemoi.models.schemas.models.NoiseInjectorSchema attribute)
(anemoi.models.schemas.models.NoOpNoiseInjectorSchema attribute)
(anemoi.training.schemas.dataloader.DataLoadingPolicySchema attribute)
(anemoi.training.schemas.training.BaseLossSchema attribute)
(anemoi.training.schemas.training.CombinedLossSchema attribute)
(anemoi.training.schemas.training.EnsembleTrainingMethodSchema attribute)
(anemoi.training.schemas.training.GenericSchema attribute)
(anemoi.training.schemas.training.SingleDeviceStrategySchema attribute)
(anemoi.training.schemas.training.SingleTrainingMethodSchema attribute)
(anemoi.training.schemas.training.TransportTrainingMethodSchema attribute)
target_datasets (anemoi.models.schemas.models.DecodersSchema attribute)
target_name (anemoi.graphs.schemas.base_graph.EdgeSchema attribute)
target_node_features (anemoi.models.schemas.models.DecodersSchema attribute)
TargetForcing (class in anemoi.training.schemas.training)
task (anemoi.training.schemas.base_schema.BaseSchema attribute)
tendencies (anemoi.training.schemas.training.UpdateDsStatsOnCkptLoadSchema attribute)
TendencyScalerSchema (class in anemoi.training.schemas.training)
TendencyScalerTargets (class in anemoi.training.schemas.training)
tensorboard (anemoi.training.schemas.system.Logs attribute)
test (anemoi.training.schemas.dataloader.DataLoaderSchema attribute)
(anemoi.training.schemas.dataloader.LoaderSet attribute)
time_aggregation_types (anemoi.training.schemas.training.TimeAggregateLossWrapperSchema attribute)
time_fraction (anemoi.training.schemas.training.TargetForcing attribute)
TimeAggregateLossWrapperSchema (class in anemoi.training.schemas.training)
timestep (anemoi.training.schemas.training.TendencyScalerSchema attribute)
TimeStepScalerSchema (class in anemoi.training.schemas.training)
training (anemoi.training.schemas.dataloader.DataLoaderSchema attribute)
(anemoi.training.schemas.dataloader.LoaderSet attribute)
training_condition (anemoi.models.schemas.models.TransportConfig attribute)
training_loss (anemoi.training.schemas.training.BaseTrainingSchema attribute)
trajectory (anemoi.training.schemas.dataloader.TrajectoryDatasetSchema attribute)
trajectory_sampling (anemoi.training.schemas.dataloader.DataLoaderSchema attribute)
TrajectoryDataset (class in anemoi.training.data.data_reader)
TrajectoryDatasetSchema (class in anemoi.training.schemas.dataloader)
TrajectorySamplingSchema (class in anemoi.training.schemas.dataloader)
TrajectorySchema (class in anemoi.training.schemas.dataloader)
transform (anemoi.training.schemas.training.SpectralLossSchema attribute)
transport (anemoi.models.schemas.models.TransportModel attribute)
(anemoi.training.schemas.training.TransportTrainingSchema attribute)
TransportConfig (class in anemoi.models.schemas.models)
TransportModel (class in anemoi.models.schemas.models)
TransportModelSchema (class in anemoi.models.schemas.models)
TransportSourceConfig (class in anemoi.models.schemas.models)
TransportTendModelSchema (class in anemoi.models.schemas.models)
TransportTrainingConfigSchema (class in anemoi.training.schemas.training)
TransportTrainingMethodSchema (class in anemoi.training.schemas.training)
TransportTrainingSchema (class in anemoi.training.schemas.training)
truncation (anemoi.training.schemas.system.InputSchema attribute)
truncation_inv (anemoi.training.schemas.system.InputSchema attribute)
U
UniformTimeStepScalerSchema (class in anemoi.training.schemas.training)
update_ds_stats_on_ckpt_load (anemoi.training.schemas.training.BaseTrainingSchema attribute)
update_scaler() (anemoi.training.losses.base.BaseLoss method)
(anemoi.training.losses.base.BaseLossWrapper method)
(anemoi.training.losses.combined.CombinedLoss method)
UpdateDsStatsOnCkptLoadSchema (class in anemoi.training.schemas.training)
upscale_mapper (anemoi.models.schemas.models.HierarchicalModelSchema attribute)
use_local_synchronization (anemoi.training.schemas.training.BaseDDPStrategySchema attribute)
use_processors_tendencies (anemoi.training.schemas.training.NaNMaskScalerSchema attribute)
V
val (anemoi.training.schemas.training.GradientClip attribute)
validate_plot_fn() (in module anemoi.training.diagnostics.evaluation.plotting.protocols)
validation (anemoi.training.schemas.dataloader.DataLoaderSchema attribute)
(anemoi.training.schemas.dataloader.LoaderSet attribute)
validation_metrics (anemoi.training.schemas.training.BaseTrainingSchema attribute)
variable_groups (anemoi.training.schemas.training.BaseTrainingSchema attribute)
VariableLevelScalerSchema (class in anemoi.training.schemas.training)
VariableLevelScalerTargets (class in anemoi.training.schemas.training)
VariableMaskingScalerSchema (class in anemoi.training.schemas.training)
variables (anemoi.training.data.data_reader.BaseAnemoiReader property)
(anemoi.training.data.multidataset.MultiDataset property)
(anemoi.training.schemas.training.VariableMaskingScalerSchema attribute)
W
wandb (anemoi.training.schemas.system.Logs attribute)
weight_averaging (anemoi.training.schemas.training.BaseTrainingSchema attribute)
weight_frac_of_total (anemoi.training.schemas.training.ReweightedGraphNodeAttributeScalerSchema attribute)
WeightAveragingSchema (class in anemoi.training.schemas.training)
weights (anemoi.training.schemas.training.GeneralVariableLossScalerSchema attribute)
(anemoi.training.schemas.training.TimeStepScalerSchema attribute)
Y
y_intercept (anemoi.training.schemas.training.VariableLevelScalerSchema attribute)