Check Gpu • AI Agents Framework

check_gpu: API reference for TrainModel.check_gpu

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

• TrainModel.run
• TrainVisionModel.run

Source

View on GitHub

praisonai/train.py at line 131