tabensemb.model.TorchModel.device

property

property TorchModel.device

The device set in the linked Trainer.

Returns:

str

“cpu” or “cuda”