tabensemb.model.AbstractModel.device

property

property AbstractModel.device

The device set in the linked Trainer.

Returns:

str

“cpu” or “cuda”