Skip to main content

check_gpu

Method
This is a method of the TrainModel class in the train module.

Signature

def check_gpu() -> Any

Returns

Returns
Any
The result of the operation.

Uses

  • get_device_properties

Used By

Source

View on GitHub

praisonai/train.py at line 131