Esc
Ask AIAnswers may be inaccurate; check the linked pages.Esc
Ask anything about these docs, like how to get started or what a function does.

neuralforecast_lightning_device

PyTorch Lightning trainer options for NeuralForecast models (TFT, TimesNet, etc.).

resolve_neuralforecast_trainer_kwargs(requested_device)

Map settings device to Lightning Trainer kwargs for NeuralForecast.

Parameter Type Default Description
requested_device Optional[str] - (undocumented)

Returns: Dict[str, Union[str, int]] (Lightning Trainer kwargs.) Raises: * RuntimeError: If requested_device is "cuda" but CUDA is unavailable. * ValueError: If requested_device is not 'cpu' or 'cuda'.