search

Deci-AI / super-gradients

Easily train or fine-tune SOTA computer vision models with one open source training library. The home of Yolo-NAS.
https://www.supergradients.com
Apache License 2.0
4.59k stars 510 forks source link

How can I enable multigpu training #921

Closed Nuwan1654 closed 1 year ago

Nuwan1654 commented 1 year ago

Is your feature request related to a problem? Please describe.

A clear and concise description of what the problem is. Ex. I'm always frustrated when [...]

Describe the solution you'd like

A clear and concise description of what you want to happen.

Describe alternatives you've considered

A clear and concise description of any alternative solutions or features you've considered.

Additional context

Add any other context or screenshots about the feature request here.

dagshub[bot] commented 1 year ago

Join the discussion on DagsHub!

BloodAxe commented 1 year ago

You can either edit recipe YAML to use multi GPU: https://github.com/Deci-AI/super-gradients/src/super_gradients/recipes/roboflow_yolo_nas_m.yaml

multi_gpu: DDP
num_gpus: 8

Or pass them via command line:

python -m super_gradients.train_from_recipe --config-name=roboflow_yolo_nas_m dataset_name=digits-t2eg6 multi_gpu=DDP num_gpus=8
Nuwan1654 commented 1 year ago

are there any recipes for yolo nas large model