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.48k stars 2.18k 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 10 months ago

Added to to list of to do in next update