ForecastingModule#

class ForecastingModule[source]#

Bases: LightningModule

Base class of all forecasting modules.

Methods

calculate_loss

compute_metrics

decode

decode_eval

decode_train

encode

forward

forward_metrics

make_chunk_specs

on_test_epoch_end

on_train_epoch_end

on_validation_epoch_end

reset_metrics

test_step

training_step

update_metrics

validation_step

Attributes

decoding_length

encoding_length

Encoding length.

head

heads

property encoding_length: int#

Encoding length.