SuperGradients is an open source neural network training library for PyTorched based computer vision models.
SuperGradients is an open source neural network training library for PyTorched based computer vision models. With SuperGradients you can easily train or fine-tune pre-trained SOTA computer vision models for object detection, image classification and semantic segmentation tasks using one training script.
https://www.supergradients.com/