vak.common.accelerator.get_default

vak.common.accelerator.get_default() → str

Get default accelerator for lightning.pytorch.Trainer.

Returns:

accelerator – Will be 'gpu' if torch.cuda.is_available() is True, and 'cpu' if not.

Return type:

str