search

OlafenwaMoses / ImageAI

A python library built to empower developers to build applications and systems with self-contained Computer Vision capabilities
https://www.genxr.co/#products
MIT License
8.64k stars 2.19k forks source link

Multiple GPU support with pytorch backend #817

Open alexpavlidespyrra opened 1 year ago

alexpavlidespyrra commented 1 year ago

It seems you have implemented multiple gpus for tensorflow backend:

from imageai.Detection.Custom import DetectionModelTrainer trainer = DetectionModelTrainer() trainer.setModelTypeAsYOLOv3() trainer.setDataDirectory(data_directory="/content/gdrive/MyDrive/ai-training/test") trainer.setGpuUsage(1) trainer.setTrainConfig(object_names_array=["Item 1", "Sample 2"], batch_size=4, num_experiments=200) trainer.trainModel()

However, this has not yet been implemented for the pytorch backend. Is this something you intend to add in future?

Thank you!

OlafenwaMoses commented 1 year ago

Added to to list of to do in next update