Deci-AI
/
super-gradients
connected to https://github.com/Deci-AI/super-gradients.git


  
1

	
2

	
3

	
4

	
5

	
6

	
7

	
8

	
            from super_gradients.training import ARCHITECTURES, losses, utils, datasets_utils, DataAugmentation, \
    TestDatasetInterface, SegmentationTestDatasetInterface, DetectionTestDatasetInterface, ClassificationTestDatasetInterface, SgModel
from super_gradients.common import init_trainer, is_distributed

__all__ = [ARCHITECTURES, losses, utils, datasets_utils, DataAugmentation,
           TestDatasetInterface, SgModel, SegmentationTestDatasetInterface, DetectionTestDatasetInterface,
           ClassificationTestDatasetInterface, init_trainer, is_distributed]