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.54k stars 496 forks source link

FILE TRAIN.py #1349

Closed chelsea456 closed 1 year ago

chelsea456 commented 1 year ago

🚀 Feature Request

hi sir!! i want to have a file python train.py such as yolo5

Proposed Solution (Optional)

No response

BloodAxe commented 1 year ago

The SuperGradients is a much more just a train.py to train yet another yolo. We provide support for various tasks and different models.
That, surely, has some learning curve. Take a time reading some docs & colab notebooks we've made

I hope after reading them it would be more clear where to start with training modes using SG.