tabensemb.model.PytorchTabular.device

property

property PytorchTabular.device

The device set in the linked Trainer.

Returns:

str

“cpu” or “cuda”