super-gradients

Easily train or fine-tune SOTA computer vision models with one open source training library. The home of Yolo-NAS.

APACHE-2.0 License

Downloads
38K
Stars
4.3K
Committers
60
super-gradients - 1.5.1

Published by deci-services over 2 years ago

  • Minor fixes for transfer learning example notebook support.
super-gradients - 1.5.0

Published by deci-services almost 3 years ago

What’s new?

  • STDC family - new recipes added with even higher mIoU💪

  • Google Colab notebook for transfer learning / fine-tuning (COCO pre-trained YOLOv5 nano into PASCAL VOC sub dataset) - Check it out in our GitHub repo README.md

  • Factories for yaml string interpolation

super-gradients - 1.4.0

Published by deci-services almost 3 years ago

  • Transfer learning example added for YoloV5.
  • Optional checkpoint root directory path argument added to SgModel.
  • Regnet recipes updated.
super-gradients - 1.3.1

Published by deci-services almost 3 years ago

Checkpoints root directory fix- allowing users to explicitly define checkpoints root directory instead of depriving it from content root.

super-gradients - 1.3.0

Published by deci-services almost 3 years ago

  • Upload to pipy added to ci workflow.
  • Max epochs limitation removed.
  • Post training phase moved before closing loggers
super-gradients - 1.2.0

Published by deci-services almost 3 years ago

  • Added an option to upload any file from the checkpoint dir.
  • Added a new callback phase and calling it on best metric.
  • Automatic GPU mode for training recipes.
  • Readme updates.
  • Sphinx documentation updates.
super-gradients - https://github.com/Deci-AI/super-gradients/releases/tag/1.1.0

Published by deci-services almost 3 years ago

This GitHub Release was done automatically by CircleCI

super-gradients - 1.0.1

Published by ranrubin almost 3 years ago

super-gradients -

Published by shaydeci almost 3 years ago

super-gradients -

Published by shaydeci almost 3 years ago

-Pretrained classification models with head replacement
-Quickstart notebook added
-Optionally loading EMA net for testing

Package Rankings
Top 3.57% on Pypi.org
Badges
Extracted from project README
Tweet Open In Colab Open In Colab Open In Colab Open In Colab Open In Colab Open In Colab Open In Colab Open In Colab Open In Colab Open In Colab Open In Colab Open In Colab Open In Colab Open In Colab Open In Colab Open In Colab Open In Colab Open In Colab Open In Colab DOI
Related Projects