Index

A

add() (anemoi.training.losses.utils.ScaleTensor method) add_scalar() (anemoi.training.losses.utils.ScaleTensor method) (anemoi.training.losses.weightedloss.BaseWeightedLoss method) allgather_batch() (anemoi.training.train.forecaster.GraphForecaster method) anemoi.training.data.dataset module anemoi.training.diagnostics.callbacks.checkpoint module anemoi.training.diagnostics.callbacks.evaluation module anemoi.training.diagnostics.callbacks.optimiser module anemoi.training.diagnostics.callbacks.plot module

anemoi.training.diagnostics.callbacks.provenance module anemoi.training.losses.combined module anemoi.training.losses.utils module anemoi.training.losses.weightedloss module anemoi.training.train.forecaster module anemoi.training.train.train module AnemoiCheckpoint (class in anemoi.training.diagnostics.callbacks.checkpoint) AnemoiTrainer (class in anemoi.training.train.train)

B

BasePerBatchPlotCallback (class in anemoi.training.diagnostics.callbacks.plot) BasePerEpochPlotCallback (class in anemoi.training.diagnostics.callbacks.plot)

BasePlotAdditionalMetrics (class in anemoi.training.diagnostics.callbacks.plot) BasePlotCallback (class in anemoi.training.diagnostics.callbacks.plot) BaseWeightedLoss (class in anemoi.training.losses.weightedloss)

C

calculate_difference() (anemoi.training.losses.weightedloss.FunctionalWeightedLoss method) calculate_val_metrics() (anemoi.training.train.forecaster.GraphForecaster method)

CombinedLoss (class in anemoi.training.losses.combined) configure_optimizers() (anemoi.training.train.forecaster.GraphForecaster method)

D

data_indices (anemoi.training.train.train.AnemoiTrainer property)

datamodule (anemoi.training.train.train.AnemoiTrainer property)

F

forward() (anemoi.training.losses.combined.CombinedLoss method) (anemoi.training.losses.weightedloss.BaseWeightedLoss method) (anemoi.training.losses.weightedloss.FunctionalWeightedLoss method) (anemoi.training.train.forecaster.GraphForecaster method)

freeze_state() (anemoi.training.losses.utils.ScaleTensor method) FunctionalWeightedLoss (class in anemoi.training.losses.weightedloss)

G

get_loss_function() (anemoi.training.train.forecaster.GraphForecaster static method) get_scalar() (anemoi.training.losses.utils.ScaleTensor method) grad_scaler() (in module anemoi.training.losses.utils)

graph_data (anemoi.training.train.train.AnemoiTrainer property) GraphForecaster (class in anemoi.training.train.forecaster) GraphTrainableFeaturesPlot (class in anemoi.training.diagnostics.callbacks.plot)

I

initial_seed (anemoi.training.train.train.AnemoiTrainer property)

L

last_checkpoint (anemoi.training.train.train.AnemoiTrainer property) LearningRateMonitor (class in anemoi.training.diagnostics.callbacks.optimiser)

LongRolloutPlots (class in anemoi.training.diagnostics.callbacks.plot) lr_scheduler_step() (anemoi.training.train.forecaster.GraphForecaster method)

M

metadata (anemoi.training.data.dataset.NativeGridDataset property) (anemoi.training.train.train.AnemoiTrainer property) mlflow_logger (anemoi.training.train.train.AnemoiTrainer property) model (anemoi.training.train.train.AnemoiTrainer property) module anemoi.training.data.dataset anemoi.training.diagnostics.callbacks.checkpoint anemoi.training.diagnostics.callbacks.evaluation anemoi.training.diagnostics.callbacks.optimiser anemoi.training.diagnostics.callbacks.plot anemoi.training.diagnostics.callbacks.provenance anemoi.training.losses.combined anemoi.training.losses.utils anemoi.training.losses.weightedloss anemoi.training.train.forecaster anemoi.training.train.train

N

name (anemoi.training.losses.weightedloss.BaseWeightedLoss property)

name_to_index (anemoi.training.data.dataset.NativeGridDataset property) NativeGridDataset (class in anemoi.training.data.dataset)

O

on_load_checkpoint() (anemoi.training.diagnostics.callbacks.provenance.ParentUUIDCallback method) on_train_end() (anemoi.training.diagnostics.callbacks.checkpoint.AnemoiCheckpoint method) on_train_epoch_end() (anemoi.training.train.forecaster.GraphForecaster method)

on_validation_batch_end() (anemoi.training.diagnostics.callbacks.evaluation.RolloutEval method) (anemoi.training.diagnostics.callbacks.plot.BasePerBatchPlotCallback method) (anemoi.training.diagnostics.callbacks.plot.LongRolloutPlots method) on_validation_epoch_end() (anemoi.training.diagnostics.callbacks.plot.BasePerEpochPlotCallback method)

P

ParentUUIDCallback (class in anemoi.training.diagnostics.callbacks.provenance) per_worker_init() (anemoi.training.data.dataset.NativeGridDataset method) PlotHistogram (class in anemoi.training.diagnostics.callbacks.plot)

PlotLoss (class in anemoi.training.diagnostics.callbacks.plot) PlotSample (class in anemoi.training.diagnostics.callbacks.plot) PlotSpectrum (class in anemoi.training.diagnostics.callbacks.plot) profiler (anemoi.training.train.train.AnemoiTrainer property)

R

remove_scalar() (anemoi.training.losses.utils.ScaleTensor method) resolution (anemoi.training.data.dataset.NativeGridDataset property) resolve() (anemoi.training.losses.utils.ScaleTensor method)

rollout_step() (anemoi.training.train.forecaster.GraphForecaster method) RolloutEval (class in anemoi.training.diagnostics.callbacks.evaluation) run_id (anemoi.training.train.train.AnemoiTrainer property)

S

scale() (anemoi.training.losses.utils.ScaleTensor method) (anemoi.training.losses.weightedloss.BaseWeightedLoss method) scale_by_node_weights() (anemoi.training.losses.weightedloss.BaseWeightedLoss method) ScaleTensor (class in anemoi.training.losses.utils) set_comm_group_info() (anemoi.training.data.dataset.NativeGridDataset method) shape (anemoi.training.losses.utils.ScaleTensor property) Shape (class in anemoi.training.losses.utils) sort_and_color_by_parameter_group (anemoi.training.diagnostics.callbacks.plot.PlotLoss property)

start_event_loop() (anemoi.training.diagnostics.callbacks.plot.BasePlotCallback method) statistics (anemoi.training.data.dataset.NativeGridDataset property) StochasticWeightAveraging (class in anemoi.training.diagnostics.callbacks.optimiser) strategy (anemoi.training.train.train.AnemoiTrainer property) submit_plot() (anemoi.training.diagnostics.callbacks.plot.BasePlotCallback method) subset() (anemoi.training.losses.utils.ScaleTensor method) subset_by_dim() (anemoi.training.losses.utils.ScaleTensor method) subset_by_str() (anemoi.training.losses.utils.ScaleTensor method) supporting_arrays (anemoi.training.data.dataset.NativeGridDataset property)

T

teardown() (anemoi.training.diagnostics.callbacks.plot.BasePlotCallback method) tensorboard_logger (anemoi.training.train.train.AnemoiTrainer property) to() (anemoi.training.losses.utils.ScaleTensor method)

train() (anemoi.training.train.train.AnemoiTrainer method) training_step() (anemoi.training.train.forecaster.GraphForecaster method) training_weights_for_imputed_variables() (anemoi.training.train.forecaster.GraphForecaster method)

U

update() (anemoi.training.losses.utils.ScaleTensor method)

update_scalar() (anemoi.training.losses.utils.ScaleTensor method) (anemoi.training.losses.weightedloss.BaseWeightedLoss method)

V

valid_date_indices (anemoi.training.data.dataset.NativeGridDataset property)

validate_scalar() (anemoi.training.losses.utils.ScaleTensor method) validation_step() (anemoi.training.train.forecaster.GraphForecaster method)

W

wandb_logger (anemoi.training.train.train.AnemoiTrainer property) without() (anemoi.training.losses.utils.ScaleTensor method)

without_by_dim() (anemoi.training.losses.utils.ScaleTensor method) without_by_str() (anemoi.training.losses.utils.ScaleTensor method) worker_init_func() (in module anemoi.training.data.dataset)