vak.common.accelerator.get_default¶
- vak.common.accelerator.get_default() str [source]¶
Get default accelerator for
lightning.pytorch.Trainer
.- Returns:
accelerator – Will be
'gpu'
iftorch.cuda.is_available()
isTrue
, and'cpu'
if not.- Return type: