Index

_

__init__() (rtnn.dataset.DataPreprocessor method) (rtnn.evaluater.MetricTracker method) (rtnn.evaluater.NMAELoss method) (rtnn.evaluater.NMSELoss method) (rtnn.logger.Logger method) (rtnn.model_utils.ModelUtils method) (rtnn.models.DimChangeModule.DimChange method) (rtnn.models.fcn.FCBlock method) (rtnn.models.fcn.FCN method) (rtnn.models.rnn.BaseRNN method) (rtnn.models.rnn.RNN_GRU method) (rtnn.models.rnn.RNN_LSTM method) (rtnn.models.Transformer.Encoder method) (rtnn.models.Transformer.EncoderTorch method) (rtnn.models.Transformer.SelfAttention method) (rtnn.models.Transformer.TransformerBlock method) (rtnn.utils.FileUtils method)

A

attention (rtnn.models.Transformer.TransformerBlock attribute)

B

BaseRNN (class in rtnn.models.rnn)

bn (rtnn.models.fcn.FCBlock attribute)

C

calc_abs() (in module rtnn.evaluater) calc_hr() (in module rtnn.evaluater) clear_logs() (rtnn.logger.Logger method) cosz (rtnn.dataset.DataPreprocessor attribute)

count (rtnn.evaluater.MetricTracker attribute) count_parameters_by_layer() (rtnn.model_utils.ModelUtils static method) create_datasets_and_loaders() (in module rtnn.main) create_normalization_mapping() (in module rtnn.main)

D

DataPreprocessor (class in rtnn.dataset) dfs (rtnn.dataset.DataPreprocessor attribute) dim_change (rtnn.models.fcn.FCN attribute)

dim_expand (rtnn.models.fcn.FCN attribute) DimChange (class in rtnn.models.DimChangeModule) dropout (rtnn.models.Transformer.Encoder attribute) (rtnn.models.Transformer.TransformerBlock attribute)

E

EasyDict (class in rtnn.utils) embed_size (rtnn.models.Transformer.Encoder attribute) (rtnn.models.Transformer.SelfAttention attribute) Encoder (class in rtnn.models.Transformer)

EncoderTorch (class in rtnn.models.Transformer) error() (rtnn.logger.Logger method) etime (rtnn.dataset.DataPreprocessor attribute) exception() (rtnn.logger.Logger method)

F

fc_out (rtnn.models.Transformer.SelfAttention attribute) FCBlock (class in rtnn.models.fcn) FCN (class in rtnn.models.fcn) feature_channel (rtnn.models.fcn.FCN attribute) feed_forward (rtnn.models.Transformer.TransformerBlock attribute) FileUtils (class in rtnn.utils) final (rtnn.models.rnn.BaseRNN attribute) (rtnn.models.Transformer.Encoder attribute) first (rtnn.models.Transformer.Encoder attribute) first_act (rtnn.models.Transformer.Encoder attribute)

forward() (rtnn.evaluater.NMAELoss method) (rtnn.evaluater.NMSELoss method) (rtnn.models.DimChangeModule.DimChange method) (rtnn.models.fcn.FCBlock method) (rtnn.models.fcn.FCN method) (rtnn.models.rnn.BaseRNN method) (rtnn.models.Transformer.Encoder method) (rtnn.models.Transformer.EncoderTorch method) (rtnn.models.Transformer.SelfAttention method) (rtnn.models.Transformer.TransformerBlock method)

G

get_data_files() (in module rtnn.main) get_loss_function() (in module rtnn.evaluater) get_parameter_number() (rtnn.model_utils.ModelUtils static method) get_version() (in module rtnn.version)

getmean() (rtnn.evaluater.MetricTracker method) getsqrtmean() (rtnn.evaluater.MetricTracker method) getstd() (rtnn.evaluater.MetricTracker method) gmrae_all() (in module rtnn.evaluater)

H

head_dim (rtnn.models.Transformer.SelfAttention attribute) heads (rtnn.models.Transformer.SelfAttention attribute)

hidden_layers (rtnn.models.fcn.FCN attribute) hidden_size (rtnn.models.rnn.BaseRNN attribute)

I

info() (rtnn.logger.Logger method) init_hidden() (rtnn.models.rnn.BaseRNN method)

initialize_model() (in module rtnn.main) input_layer (rtnn.models.fcn.FCN attribute)

K

keys (rtnn.models.Transformer.SelfAttention attribute)

L

lai (rtnn.dataset.DataPreprocessor attribute) layers (rtnn.models.Transformer.Encoder attribute) linear (rtnn.models.fcn.FCBlock attribute) load_checkpoint() (rtnn.model_utils.ModelUtils static method) load_checkpoint_if_requested() (in module rtnn.main) load_model() (in module rtnn.model_loader)

load_training_checkpoint() (rtnn.model_utils.ModelUtils static method) log_configuration() (in module rtnn.main) log_metrics() (rtnn.logger.Logger method) log_model_summary() (rtnn.model_utils.ModelUtils static method) Logger (class in rtnn.logger) logger (rtnn.dataset.DataPreprocessor attribute)

M

mae_all() (in module rtnn.evaluater) main() (in module rtnn.main) makedir() (rtnn.utils.FileUtils static method) makefile() (rtnn.utils.FileUtils static method) mare_all() (in module rtnn.evaluater) mbe_all() (in module rtnn.evaluater) MetricTracker (class in rtnn.evaluater) min_dims (rtnn.dataset.DataPreprocessor attribute) ModelUtils (class in rtnn.model_utils) module rtnn.dataset rtnn.diagnostics rtnn.evaluater rtnn.logger rtnn.main rtnn.model_loader rtnn.model_utils rtnn.models.DimChangeModule rtnn.models.fcn rtnn.models.rnn rtnn.models.Transformer rtnn.utils rtnn.version

mse_all() (in module rtnn.evaluater)

N

nmae_all() (in module rtnn.evaluater) NMAELoss (class in rtnn.evaluater) nmse_all() (in module rtnn.evaluater) NMSELoss (class in rtnn.evaluater) norm1 (rtnn.models.Transformer.TransformerBlock attribute)

norm2 (rtnn.models.Transformer.TransformerBlock attribute) norm_mapping (rtnn.dataset.DataPreprocessor attribute) normalization_type (rtnn.dataset.DataPreprocessor attribute) normalize() (rtnn.dataset.DataPreprocessor method) num_layers (rtnn.models.rnn.BaseRNN attribute)

O

output_channel (rtnn.models.fcn.FCN attribute) (rtnn.models.rnn.BaseRNN attribute)

output_layer (rtnn.models.fcn.FCN attribute) ov (rtnn.dataset.DataPreprocessor attribute)

P

parse_args() (in module rtnn.main) parse_years() (in module rtnn.main) plot_flux_and_abs() (in module rtnn.diagnostics) plot_flux_and_abs_lines() (in module rtnn.diagnostics) plot_HeatRate() (in module rtnn.diagnostics) plot_loss_histories() (in module rtnn.diagnostics)

plot_metric_histories() (in module rtnn.diagnostics) plot_RTM() (in module rtnn.diagnostics) plot_spatial_temporal_density() (in module rtnn.diagnostics) position_embedding (rtnn.models.Transformer.Encoder attribute) print_model_layers() (rtnn.model_utils.ModelUtils static method) print_version() (in module rtnn.main)

Q

queries (rtnn.models.Transformer.SelfAttention attribute)

R

r2_all() (in module rtnn.evaluater) relu (rtnn.models.fcn.FCBlock attribute) reset() (rtnn.evaluater.MetricTracker method) rnn (rtnn.models.rnn.BaseRNN attribute) RNN_GRU (class in rtnn.models.rnn) RNN_LSTM (class in rtnn.models.rnn) rs (rtnn.dataset.DataPreprocessor attribute) rtnn.dataset module rtnn.diagnostics module rtnn.evaluater module rtnn.logger module rtnn.main module

rtnn.model_loader module rtnn.model_utils module rtnn.models.DimChangeModule module rtnn.models.fcn module rtnn.models.rnn module rtnn.models.Transformer module rtnn.utils module rtnn.version module run_validation() (in module rtnn.evaluater)

S

save_checkpoint() (in module rtnn.main) (rtnn.model_utils.ModelUtils static method) save_emergency_checkpoint() (rtnn.model_utils.ModelUtils static method) save_training_checkpoint() (rtnn.model_utils.ModelUtils static method) sbatch (rtnn.dataset.DataPreprocessor attribute) SelfAttention (class in rtnn.models.Transformer) seq_length (rtnn.models.fcn.FCN attribute) (rtnn.models.Transformer.Encoder attribute)

setup_device_and_seed() (in module rtnn.main) setup_directories_and_logging() (in module rtnn.main) show_header() (rtnn.logger.Logger method) ssa (rtnn.dataset.DataPreprocessor attribute) start_task() (rtnn.logger.Logger method) stats() (in module rtnn.diagnostics) step() (rtnn.logger.Logger method) stime (rtnn.dataset.DataPreprocessor attribute) success() (rtnn.logger.Logger method)

T

tbatch (rtnn.dataset.DataPreprocessor attribute) time_blocks (rtnn.dataset.DataPreprocessor attribute)

train_epoch() (in module rtnn.main) TransformerBlock (class in rtnn.models.Transformer) tstep (rtnn.dataset.DataPreprocessor attribute)

U

unnorm_mpas() (in module rtnn.evaluater)

update() (rtnn.evaluater.MetricTracker method)

V

value (rtnn.evaluater.MetricTracker attribute)

values (rtnn.models.Transformer.SelfAttention attribute)

W

warning() (rtnn.logger.Logger method)

Y

years (rtnn.dataset.DataPreprocessor attribute)