Contents:
get_model()
Get model object, depending on the run configuration.
cfg (Config) – The run configuration.
A new model instance of the type specified in the config.
nn.Module
ARLSTM
ARLSTM.forward()
BaseModel
BaseModel.forward()
BaseModel.module_parts
BaseModel.pre_model_hook()
BaseModel.sample()
CudaLSTM
CudaLSTM.forward()
CudaLSTM.module_parts
CustomLSTM
CustomLSTM.copy_weights()
CustomLSTM.forward()
EALSTM
EALSTM.forward()
EALSTM.module_parts
EmbCudaLSTM
EmbCudaLSTM.forward()
EmbCudaLSTM.module_parts
FC
FC.forward()
GRU
GRU.forward()
GRU.module_parts
HandoffForecastLSTM
HandoffForecastLSTM.forward()
HandoffForecastLSTM.module_parts
CMAL
CMAL.forward()
GMM
GMM.forward()
Regression
Regression.forward()
UMAL
UMAL.forward()
get_head()
InputLayer
InputLayer.forward()
Mamba
Mamba.forward()
Mamba.module_parts
MCLSTM
MCLSTM.forward()
MCLSTM.module_parts
MultiHeadForecastLSTM
MultiHeadForecastLSTM.forward()
MultiHeadForecastLSTM.module_parts
MTSLSTM
MTSLSTM.forward()
MTSLSTM.module_parts
ODELSTM
ODELSTM.forward()
ODELSTM.module_parts
SequentialForecastLSTM
SequentialForecastLSTM.forward()
SequentialForecastLSTM.module_parts
StackedForecastLSTM
StackedForecastLSTM.forward()
StackedForecastLSTM.module_parts
TemplateModel
TemplateModel.forward()
TemplateModel.module_parts
Transformer
Transformer.forward()
Transformer.module_parts