tabensemb.model.AbstractNN.on_gpu

property

property AbstractNN.on_gpu: bool

Returns True if this model is currently located on a GPU.

Useful to set flags around the LightningModule for different CPU vs GPU behavior.