Register
Login
Resources
Docs Blog Datasets Glossary Case Studies Tutorials & Webinars
Product
Data Engine LLMs Platform Enterprise
Pricing Explore
Connect to our Discord channel
Resources  
Docs Blog Datasets Glossary Case Studies Tutorials & Webinars
Explore Pricing
Product  
Data Engine LLMs Platform Enterprise
Connect to Dagshub Discord channel
Register
Login

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

Watch 4
Issues 62 Pull Requests 44 Discussions 0

deci_core_unit_test_suite_runner.py 9.7 KB
History Raw

You have to be logged in to leave a comment. Sign In
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149 
  1. import sys
    2. 

  2. import unittest
    3. 

  3. 

  4. from tests.integration_tests.ema_train_integration_test import EMAIntegrationTest
    5. 

  5. from tests.unit_tests import (
    6. 

  6.     ZeroWdForBnBiasTest,
    7. 

  7.     SaveCkptListUnitTest,
    8. 

  8.     TestAverageMeter,
    9. 

  9.     TestRepVgg,
    10. 

  10.     TestWithoutTrainTest,
    11. 

  11.     OhemLossTest,
    12. 

  12.     EarlyStopTest,
    13. 

  13.     SegmentationTransformsTest,
    14. 

  14.     TestConvBnRelu,
    15. 

  15.     FactoriesTest,
    16. 

  16.     InitializeWithDataloadersTest,
    17. 

  17.     TrainingParamsTest,
    18. 

  18.     TrainOptimizerParamsOverride,
    19. 

  19.     CallTrainTwiceTest,
    20. 

  20.     ResumeTrainingTest,
    21. 

  21.     CallTrainAfterTestTest,
    22. 

  22.     CrashTipTest,
    23. 

  23.     TestTransforms,
    24. 

  24. )
    25. 

  25. from tests.end_to_end_tests import TestTrainer
    26. 

  26. from tests.unit_tests.detection_utils_test import TestDetectionUtils
    27. 

  27. from tests.unit_tests.detection_dataset_test import DetectionDatasetTest
    28. 

  28. from tests.unit_tests.export_onnx_test import TestModelsONNXExport
    29. 

  29. from tests.unit_tests.load_checkpoint_test import LoadCheckpointTest
    30. 

  30. from tests.unit_tests.local_ckpt_head_replacement_test import LocalCkptHeadReplacementTest
    31. 

  31. from tests.unit_tests.max_batches_loop_break_test import MaxBatchesLoopBreakTest
    32. 

  32. from tests.unit_tests.phase_delegates_test import ContextMethodsTest
    33. 

  33. from tests.unit_tests.pose_estimation_dataset_test import TestPoseEstimationDataset
    34. 

  34. from tests.unit_tests.preprocessing_unit_test import PreprocessingUnitTest
    35. 

  35. from tests.unit_tests.quantization_utility_tests import QuantizationUtilityTest
    36. 

  36. from tests.unit_tests.random_erase_test import RandomEraseTest
    37. 

  37. from tests.unit_tests.replace_head_test import ReplaceHeadUnitTest
    38. 

  38. from tests.unit_tests.strictload_enum_test import StrictLoadEnumTest
    39. 

  39. from tests.unit_tests.train_with_intialized_param_args_test import TrainWithInitializedObjectsTest
    40. 

  40. from tests.unit_tests.pretrained_models_unit_test import PretrainedModelsUnitTest
    41. 

  41. from tests.unit_tests.lr_warmup_test import LRWarmupTest
    42. 

  42. from tests.unit_tests.kd_ema_test import KDEMATest
    43. 

  43. from tests.unit_tests.kd_trainer_test import KDTrainerTest
    44. 

  44. from tests.unit_tests.dice_loss_test import DiceLossTest
    45. 

  45. from tests.unit_tests.iou_loss_test import IoULossTest
    46. 

  46. from tests.unit_tests.update_param_groups_unit_test import UpdateParamGroupsTest
    47. 

  47. from tests.unit_tests.vit_unit_test import TestViT
    48. 

  48. from tests.unit_tests.yolox_unit_test import TestYOLOX
    49. 

  49. from tests.unit_tests.lr_cooldown_test import LRCooldownTest
    50. 

  50. from tests.unit_tests.detection_targets_format_transform_test import DetectionTargetsTransformTest
    51. 

  51. from tests.unit_tests.forward_pass_prep_fn_test import ForwardpassPrepFNTest
    52. 

  52. from tests.unit_tests.mask_loss_test import MaskAttentionLossTest
    53. 

  53. from tests.unit_tests.detection_sub_sampling_test import TestDetectionDatasetSubsampling
    54. 

  54. from tests.unit_tests.detection_sub_classing_test import TestDetectionDatasetSubclassing
    55. 

  55. from tests.unit_tests.detection_output_adapter_test import TestDetectionOutputAdapter
    56. 

  56. from tests.unit_tests.detection_caching import TestDetectionDatasetCaching
    57. 

  57. from tests.unit_tests.multi_scaling_test import MultiScaleTest
    58. 

  58. from tests.unit_tests.ppyoloe_unit_test import TestPPYOLOE
    59. 

  59. from tests.unit_tests.bbox_formats_test import BBoxFormatsTest
    60. 

  60. from tests.unit_tests.config_inspector_test import ConfigInspectTest
    61. 

  61. from tests.unit_tests.repvgg_block_tests import TestRepVGGBlock
    62. 

  62. from tests.unit_tests.training_utils_test import TestTrainingUtils
    63. 

  63. from tests.unit_tests.dekr_loss_test import DEKRLossTest
    64. 

  64. from tests.unit_tests.pose_estimation_metrics_test import TestPoseEstimationMetrics
    65. 

  65. 

  66. 

  67. class CoreUnitTestSuiteRunner:
    68. 

  68.     def __init__(self):
    69. 

  69.         self.test_loader = unittest.TestLoader()
    70. 

  70.         self.unit_tests_suite = unittest.TestSuite()
    71. 

  71.         self._add_modules_to_unit_tests_suite()
    72. 

  72.         self.end_to_end_tests_suite = unittest.TestSuite()
    73. 

  73.         self._add_modules_to_end_to_end_tests_suite()
    74. 

  74.         self.test_runner = unittest.TextTestRunner(verbosity=3, stream=sys.stdout)
    75. 

  75. 

  76.     def _add_modules_to_unit_tests_suite(self):
    77. 

  77.         """
    78. 

  78.         _add_modules_to_unit_tests_suite - Adds unit tests to the Unit Tests Test Suite
    79. 

  79.             :return:
    80. 

  80.         """
    81. 

  81.         self.unit_tests_suite.addTest(self.test_loader.loadTestsFromModule(CrashTipTest))
    82. 

  82.         self.unit_tests_suite.addTest(self.test_loader.loadTestsFromModule(SaveCkptListUnitTest))
    83. 

  83.         self.unit_tests_suite.addTest(self.test_loader.loadTestsFromModule(ZeroWdForBnBiasTest))
    84. 

  84.         self.unit_tests_suite.addTest(self.test_loader.loadTestsFromModule(TestAverageMeter))
    85. 

  85.         self.unit_tests_suite.addTest(self.test_loader.loadTestsFromModule(TestRepVgg))
    86. 

  86.         self.unit_tests_suite.addTest(self.test_loader.loadTestsFromModule(TestWithoutTrainTest))
    87. 

  87.         self.unit_tests_suite.addTest(self.test_loader.loadTestsFromModule(StrictLoadEnumTest))
    88. 

  88.         self.unit_tests_suite.addTest(self.test_loader.loadTestsFromModule(TrainWithInitializedObjectsTest))
    89. 

  89.         self.unit_tests_suite.addTest(self.test_loader.loadTestsFromModule(RandomEraseTest))
    90. 

  90.         self.unit_tests_suite.addTest(self.test_loader.loadTestsFromModule(OhemLossTest))
    91. 

  91.         self.unit_tests_suite.addTest(self.test_loader.loadTestsFromModule(EarlyStopTest))
    92. 

  92.         self.unit_tests_suite.addTest(self.test_loader.loadTestsFromModule(SegmentationTransformsTest))
    93. 

  93.         self.unit_tests_suite.addTest(self.test_loader.loadTestsFromModule(PretrainedModelsUnitTest))
    94. 

  94.         self.unit_tests_suite.addTest(self.test_loader.loadTestsFromModule(LRWarmupTest))
    95. 

  95.         self.unit_tests_suite.addTest(self.test_loader.loadTestsFromModule(TestConvBnRelu))
    96. 

  96.         self.unit_tests_suite.addTest(self.test_loader.loadTestsFromModule(FactoriesTest))
    97. 

  97.         self.unit_tests_suite.addTest(self.test_loader.loadTestsFromModule(TestDetectionUtils))
    98. 

  98.         self.unit_tests_suite.addTest(self.test_loader.loadTestsFromModule(DiceLossTest))
    99. 

  99.         self.unit_tests_suite.addTest(self.test_loader.loadTestsFromModule(TestViT))
    100. 

  100.         self.unit_tests_suite.addTest(self.test_loader.loadTestsFromModule(KDEMATest))
    101. 

  101.         self.unit_tests_suite.addTest(self.test_loader.loadTestsFromModule(KDTrainerTest))
    102. 

  102.         self.unit_tests_suite.addTest(self.test_loader.loadTestsFromModule(TestYOLOX))
    103. 

  103.         self.unit_tests_suite.addTest(self.test_loader.loadTestsFromModule(InitializeWithDataloadersTest))
    104. 

  104.         self.unit_tests_suite.addTest(self.test_loader.loadTestsFromModule(LRCooldownTest))
    105. 

  105.         self.unit_tests_suite.addTest(self.test_loader.loadTestsFromModule(DetectionTargetsTransformTest))
    106. 

  106.         self.unit_tests_suite.addTest(self.test_loader.loadTestsFromModule(ForwardpassPrepFNTest))
    107. 

  107.         self.unit_tests_suite.addTest(self.test_loader.loadTestsFromModule(ContextMethodsTest))
    108. 

  108.         self.unit_tests_suite.addTest(self.test_loader.loadTestsFromModule(UpdateParamGroupsTest))
    109. 

  109.         self.unit_tests_suite.addTest(self.test_loader.loadTestsFromModule(MaskAttentionLossTest))
    110. 

  110.         self.unit_tests_suite.addTest(self.test_loader.loadTestsFromModule(IoULossTest))
    111. 

  111.         self.unit_tests_suite.addTest(self.test_loader.loadTestsFromModule(TestDetectionDatasetSubsampling))
    112. 

  112.         self.unit_tests_suite.addTest(self.test_loader.loadTestsFromModule(TestDetectionDatasetSubclassing))
    113. 

  113.         self.unit_tests_suite.addTest(self.test_loader.loadTestsFromModule(QuantizationUtilityTest))
    114. 

  114.         self.unit_tests_suite.addTest(self.test_loader.loadTestsFromModule(TestDetectionDatasetCaching))
    115. 

  115.         self.unit_tests_suite.addTest(self.test_loader.loadTestsFromModule(MultiScaleTest))
    116. 

  116.         self.unit_tests_suite.addTest(self.test_loader.loadTestsFromModule(TrainingParamsTest))
    117. 

  117.         self.unit_tests_suite.addTest(self.test_loader.loadTestsFromModule(CallTrainTwiceTest))
    118. 

  118.         self.unit_tests_suite.addTest(self.test_loader.loadTestsFromModule(TrainOptimizerParamsOverride))
    119. 

  119.         self.unit_tests_suite.addTest(self.test_loader.loadTestsFromModule(BBoxFormatsTest))
    120. 

  120.         self.unit_tests_suite.addTest(self.test_loader.loadTestsFromModule(ResumeTrainingTest))
    121. 

  121.         self.unit_tests_suite.addTest(self.test_loader.loadTestsFromModule(CallTrainAfterTestTest))
    122. 

  122.         self.unit_tests_suite.addTest(self.test_loader.loadTestsFromModule(ConfigInspectTest))
    123. 

  123.         self.unit_tests_suite.addTest(self.test_loader.loadTestsFromModule(TestDetectionOutputAdapter))
    124. 

  124.         self.unit_tests_suite.addTest(self.test_loader.loadTestsFromModule(TestRepVGGBlock))
    125. 

  125.         self.unit_tests_suite.addTest(self.test_loader.loadTestsFromModule(LocalCkptHeadReplacementTest))
    126. 

  126.         self.unit_tests_suite.addTest(self.test_loader.loadTestsFromModule(DetectionDatasetTest))
    127. 

  127.         self.unit_tests_suite.addTest(self.test_loader.loadTestsFromModule(TestModelsONNXExport))
    128. 

  128.         self.unit_tests_suite.addTest(self.test_loader.loadTestsFromModule(MaxBatchesLoopBreakTest))
    129. 

  129.         self.unit_tests_suite.addTest(self.test_loader.loadTestsFromModule(TestTrainingUtils))
    130. 

  130.         self.unit_tests_suite.addTest(self.test_loader.loadTestsFromModule(TestTransforms))
    131. 

  131.         self.unit_tests_suite.addTest(self.test_loader.loadTestsFromModule(TestPPYOLOE))
    132. 

  132.         self.unit_tests_suite.addTest(self.test_loader.loadTestsFromModule(DEKRLossTest))
    133. 

  133.         self.unit_tests_suite.addTest(self.test_loader.loadTestsFromModule(TestPoseEstimationMetrics))
    134. 

  134.         self.unit_tests_suite.addTest(self.test_loader.loadTestsFromModule(TestPoseEstimationDataset))
    135. 

  135.         self.unit_tests_suite.addTest(self.test_loader.loadTestsFromModule(LoadCheckpointTest))
    136. 

  136.         self.unit_tests_suite.addTest(self.test_loader.loadTestsFromModule(ReplaceHeadUnitTest))
    137. 

  137.         self.unit_tests_suite.addTest(self.test_loader.loadTestsFromModule(PreprocessingUnitTest))
    138. 

  138. 

  139.     def _add_modules_to_end_to_end_tests_suite(self):
    140. 

  140.         """
    141. 

  141.         _add_modules_to_end_to_end_tests_suite - Adds end to end tests to the Unit Tests Test Suite
    142. 

  142.             :return:
    143. 

  143.         """
    144. 

  144.         self.end_to_end_tests_suite.addTest(self.test_loader.loadTestsFromModule(TestTrainer))
    145. 

  145.         self.end_to_end_tests_suite.addTest(self.test_loader.loadTestsFromModule(EMAIntegrationTest))
    146. 

  146. 

  147. 

  148. if __name__ == "__main__":
    149. 

  149.     unittest.main()
    150.
Prev
Next
Tip!

Press p or to see the previous file or, n or to see the next file

Comments

Write Preview
Loading...
Integrate AWS S3
Use S3 remote
Select bucket
Access key
Finish

Use AWS S3 as storage!

Browsing data directories saved to S3 is possible with DAGsHub. Let's configure your repository to easily display your data in the context of any commit!

Specify your S3 bucket

Access key (If needed)

Congratulations!

super-gradients is now integrated with AWS S3!

Delete Storage Key

Are you sure you want to delete this access key?

No
Yes
Integrate Google Cloud Storage
Use Google Storage
Select bucket
Upload key
Finish

Use Google Cloud Storage!

Browsing data directories saved to Google Cloud Storage is possible with DAGsHub. Let's configure your repository to easily display your data in the context of any commit!

Specify your Google Storage bucket

Service Account Key

Congratulations!

super-gradients is now integrated with Google Cloud Storage!

Delete Storage Key

Are you sure you want to delete this access key?

No
Yes
Integrate Azure Cloud Storage
Use Azure Storage
Select bucket
Set key
Finish

Use Azure Cloud Storage!

Browsing data directories saved to Azure Cloud Storage is possible with DAGsHub. Let's configure your repository to easily display your data in the context of any commit!

Specify your Azure Storage bucket

Access key (If needed)

Congratulations!

super-gradients is now integrated with Azure Cloud Storage!

Delete Storage Key

Are you sure you want to delete this access key?

No
Yes
Integrate S3 compatible storage
Use S3 like remote
Select bucket
Access key
Finish

Use any S3 compatible storage!

Browsing data directories saved to S3 compatible storage is possible with DAGsHub. Let's configure your repository to easily display your data in the context of any commit!

Specify your S3 bucket

Bucket name cannot be the same as the repository name. Please change one of them.

Access key (If needed)

Congratulations!

super-gradients is now integrated with your S3 compatible storage!

Delete Storage Key

Are you sure you want to delete this access key?

No
Yes