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 7.8 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 
  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. )
    24. 

  24. from tests.end_to_end_tests import TestTrainer
    25. 

  25. from tests.unit_tests.local_ckpt_head_replacement_test import LocalCkptHeadReplacementTest
    26. 

  26. from tests.unit_tests.phase_delegates_test import ContextMethodsTest
    27. 

  27. from tests.unit_tests.quantization_utility_tests import QuantizationUtilityTest
    28. 

  28. from tests.unit_tests.random_erase_test import RandomEraseTest
    29. 

  29. from tests.unit_tests.strictload_enum_test import StrictLoadEnumTest
    30. 

  30. from tests.unit_tests.train_with_intialized_param_args_test import TrainWithInitializedObjectsTest
    31. 

  31. from tests.unit_tests.pretrained_models_unit_test import PretrainedModelsUnitTest
    32. 

  32. from tests.unit_tests.lr_warmup_test import LRWarmupTest
    33. 

  33. from tests.unit_tests.kd_ema_test import KDEMATest
    34. 

  34. from tests.unit_tests.kd_trainer_test import KDTrainerTest
    35. 

  35. from tests.unit_tests.dice_loss_test import DiceLossTest
    36. 

  36. from tests.unit_tests.iou_loss_test import IoULossTest
    37. 

  37. from tests.unit_tests.update_param_groups_unit_test import UpdateParamGroupsTest
    38. 

  38. from tests.unit_tests.vit_unit_test import TestViT
    39. 

  39. from tests.unit_tests.yolox_unit_test import TestYOLOX
    40. 

  40. from tests.unit_tests.lr_cooldown_test import LRCooldownTest
    41. 

  41. from tests.unit_tests.detection_targets_format_transform_test import DetectionTargetsTransformTest
    42. 

  42. from tests.unit_tests.forward_pass_prep_fn_test import ForwardpassPrepFNTest
    43. 

  43. from tests.unit_tests.mask_loss_test import MaskAttentionLossTest
    44. 

  44. from tests.unit_tests.detection_sub_sampling_test import TestDetectionDatasetSubsampling
    45. 

  45. from tests.unit_tests.detection_sub_classing_test import TestDetectionDatasetSubclassing
    46. 

  46. from tests.unit_tests.detection_output_adapter_test import TestDetectionOutputAdapter
    47. 

  47. from tests.unit_tests.detection_caching import TestDetectionDatasetCaching
    48. 

  48. from tests.unit_tests.multi_scaling_test import MultiScaleTest
    49. 

  49. from tests.unit_tests.ppyoloe_unit_test import PPYoloETests
    50. 

  50. from tests.unit_tests.bbox_formats_test import BBoxFormatsTest
    51. 

  51. from tests.unit_tests.config_inspector_test import ConfigInspectTest
    52. 

  52. from tests.unit_tests.repvgg_block_tests import TestRepVGGBlock
    53. 

  53. 

  54. 

  55. class CoreUnitTestSuiteRunner:
    56. 

  56.     def __init__(self):
    57. 

  57.         self.test_loader = unittest.TestLoader()
    58. 

  58.         self.unit_tests_suite = unittest.TestSuite()
    59. 

  59.         self._add_modules_to_unit_tests_suite()
    60. 

  60.         self.end_to_end_tests_suite = unittest.TestSuite()
    61. 

  61.         self._add_modules_to_end_to_end_tests_suite()
    62. 

  62.         self.test_runner = unittest.TextTestRunner(verbosity=3, stream=sys.stdout)
    63. 

  63. 

  64.     def _add_modules_to_unit_tests_suite(self):
    65. 

  65.         """
    66. 

  66.         _add_modules_to_unit_tests_suite - Adds unit tests to the Unit Tests Test Suite
    67. 

  67.             :return:
    68. 

  68.         """
    69. 

  69.         self.unit_tests_suite.addTest(self.test_loader.loadTestsFromModule(CrashTipTest))
    70. 

  70.         self.unit_tests_suite.addTest(self.test_loader.loadTestsFromModule(SaveCkptListUnitTest))
    71. 

  71.         self.unit_tests_suite.addTest(self.test_loader.loadTestsFromModule(ZeroWdForBnBiasTest))
    72. 

  72.         self.unit_tests_suite.addTest(self.test_loader.loadTestsFromModule(TestAverageMeter))
    73. 

  73.         self.unit_tests_suite.addTest(self.test_loader.loadTestsFromModule(TestRepVgg))
    74. 

  74.         self.unit_tests_suite.addTest(self.test_loader.loadTestsFromModule(TestWithoutTrainTest))
    75. 

  75.         self.unit_tests_suite.addTest(self.test_loader.loadTestsFromModule(StrictLoadEnumTest))
    76. 

  76.         self.unit_tests_suite.addTest(self.test_loader.loadTestsFromModule(TrainWithInitializedObjectsTest))
    77. 

  77.         self.unit_tests_suite.addTest(self.test_loader.loadTestsFromModule(RandomEraseTest))
    78. 

  78.         self.unit_tests_suite.addTest(self.test_loader.loadTestsFromModule(OhemLossTest))
    79. 

  79.         self.unit_tests_suite.addTest(self.test_loader.loadTestsFromModule(EarlyStopTest))
    80. 

  80.         self.unit_tests_suite.addTest(self.test_loader.loadTestsFromModule(SegmentationTransformsTest))
    81. 

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

  114. 

  115.     def _add_modules_to_end_to_end_tests_suite(self):
    116. 

  116.         """
    117. 

  117.         _add_modules_to_end_to_end_tests_suite - Adds end to end tests to the Unit Tests Test Suite
    118. 

  118.             :return:
    119. 

  119.         """
    120. 

  120.         self.end_to_end_tests_suite.addTest(self.test_loader.loadTestsFromModule(TestTrainer))
    121. 

  121.         self.end_to_end_tests_suite.addTest(self.test_loader.loadTestsFromModule(EMAIntegrationTest))
    122. 

  122. 

  123. 

  124. if __name__ == "__main__":
    125. 

  125.     unittest.main()
    126.
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