tabensemb.model.WideDeep.device

property

property WideDeep.device

The device set in the linked Trainer.

Returns:

str

“cpu” or “cuda”