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


  
1

	
2

	
3

	
4

	
5

	
6

	
7

	
8

	
9

	
            # PACKAGE IMPORTS FOR EXTERNAL USAGE
import super_gradients.training.utils.distributed_training_utils as distributed_training_utils
from super_gradients.training.datasets import datasets_utils, DataAugmentation, DetectionDataSet, TestDatasetInterface, SegmentationTestDatasetInterface, DetectionTestDatasetInterface, ClassificationTestDatasetInterface
from super_gradients.training.models import ARCHITECTURES
from super_gradients.training.sg_model import SgModel, \
    MultiGPUMode, StrictLoad

__all__ = [distributed_training_utils, datasets_utils, DataAugmentation, DetectionDataSet, TestDatasetInterface,
           ARCHITECTURES, SgModel, MultiGPUMode, TestDatasetInterface, SegmentationTestDatasetInterface, DetectionTestDatasetInterface, ClassificationTestDatasetInterface, StrictLoad]