hy2dl.modelzoo

hy2dl.modelzoo.get_model(cfg: Config) → Module

Get model object, depending on the run configuration.

Parameters:

cfg (Config) – The run configuration.

Returns:

A new model instance of the type specified in the config.

Return type:

nn.Module

• CudaLSTM
  • CudaLSTM
    • CudaLSTM.forward()
• Input Layer
  • InputLayer
    • InputLayer.build_embedding()
    • InputLayer.build_ffnn()
    • InputLayer.forward()
• LSTM mix-density-networks
  • LSTMMDN
    • LSTMMDN.forward()
• Hybrid model
  • Hybrid
    • Hybrid.forward()