tabensemb.trainer.Trainer.set_device

method

Trainer.set_device(device: str)

Set the device on which models are trained.

Parameters:

device

“cpu”, “cuda”, or “cuda:X” (if available)

Notes

Multi-GPU training and training on a machine with multiple GPUs are not tested.