ontime.core.modelling.libs.pytorch.pytorch_forecasting_model

Classes

TorchForecastingModel(torch_model, ...[, ...])

Generic wrapper around PyTorch forecasting model, using PyTorch lightning for training.