sub_modules

Implementations of nn.RNNBase for DeepAR.

Functions

get_cell(cell_type)

Classes

TimeSeriesGRU(*args, **kwargs)	Implementation of GRU module compatible with DeepAR.
TimeSeriesLSTM(*args, **kwargs)	Implementation of LSTM module compatible with DeepAR.
TimeSeriesRNN(*args, **kwargs)	Base class for implementations of RNN modules compatible with DeepAR.

DeepAR pytorch_forecasting.models.deepar.sub_modules.get_cell