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

all_architectures.py 5.3 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 
  1. from super_gradients.training.models import ResNeXt50, ResNeXt101, googlenet_v1
    2. 

  2. from super_gradients.training.models.classification_models import repvgg, efficientnet, densenet, resnet, regnet
    3. 

  3. from super_gradients.training.models.classification_models.mobilenetv2 import mobile_net_v2, mobile_net_v2_135, \
    4. 

  4.     custom_mobile_net_v2
    5. 

  5. from super_gradients.training.models.classification_models.mobilenetv3 import mobilenetv3_large, mobilenetv3_small, \
    6. 

  6.     mobilenetv3_custom
    7. 

  7. from super_gradients.training.models.classification_models.shufflenetv2 import ShufflenetV2_x0_5, ShufflenetV2_x1_0, \
    8. 

  8.     ShufflenetV2_x1_5, \
    9. 

  9.     ShufflenetV2_x2_0, CustomizedShuffleNetV2
    10. 

  10. from super_gradients.training.models.detection_models.csp_darknet53 import CSPDarknet53
    11. 

  11. from super_gradients.training.models.detection_models.darknet53 import Darknet53
    12. 

  12. from super_gradients.training.models.detection_models.ssd import SSDMobileNetV1, SSDLiteMobileNetV2
    13. 

  13. from super_gradients.training.models.detection_models.yolov3 import YoloV3, TinyYoloV3
    14. 

  14. from super_gradients.training.models.detection_models.yolov5 import YoLoV5N, YoLoV5S, YoLoV5M, YoLoV5L, YoLoV5X, Custom_YoLoV5
    15. 

  15. from super_gradients.training.models.segmentation_models.ddrnet import DDRNet23, DDRNet23Slim, AnyBackBoneDDRNet23
    16. 

  16. from super_gradients.training.models.segmentation_models.shelfnet import ShelfNet18, ShelfNet34, ShelfNet50, \
    17. 

  17.     ShelfNet503343, ShelfNet101
    18. 

  18. 

  19. # IF YOU UPDATE THE ARCHITECTURE DICT PLEASE ALSO UPDATE THE ENUM CLASS DOWN BELOW.
    20. 

  20. 

  21. 

  22. ARCHITECTURES = {"resnet18": resnet.ResNet18,
    23. 

  23.                  "resnet34": resnet.ResNet34,
    24. 

  24.                  "resnet50_3343": resnet.ResNet50_3343,
    25. 

  25.                  "resnet50": resnet.ResNet50,
    26. 

  26.                  "resnet101": resnet.ResNet101,
    27. 

  27.                  "resnet152": resnet.ResNet152,
    28. 

  28.                  "resnet18_cifar": resnet.ResNet18Cifar,
    29. 

  29.                  "custom_resnet": resnet.CustomizedResnet,
    30. 

  30.                  "custom_resnet50": resnet.CustomizedResnet50,
    31. 

  31.                  "custom_resnet_cifar": resnet.CustomizedResnetCifar,
    32. 

  32.                  "custom_resnet50_cifar": resnet.CustomizedResnet50Cifar,
    33. 

  33.                  "mobilenet_v2": mobile_net_v2,
    34. 

  34.                  "mobile_net_v2_135": mobile_net_v2_135,
    35. 

  35.                  "custom_mobilenet_v2": custom_mobile_net_v2,
    36. 

  36.                  "mobilenet_v3_large": mobilenetv3_large,
    37. 

  37.                  "mobilenet_v3_small": mobilenetv3_small,
    38. 

  38.                  "mobilenet_v3_custom": mobilenetv3_custom,
    39. 

  39.                  "yolo_v3": YoloV3,
    40. 

  40.                  "tiny_yolo_v3": TinyYoloV3,
    41. 

  41.                  "custom_densenet": densenet.CustomizedDensnet,
    42. 

  42.                  "densenet121": densenet.densenet121,
    43. 

  43.                  "densenet161": densenet.densenet161,
    44. 

  44.                  "densenet169": densenet.densenet169,
    45. 

  45.                  "densenet201": densenet.densenet201,
    46. 

  46.                  "shelfnet18": ShelfNet18,
    47. 

  47.                  "shelfnet34": ShelfNet34,
    48. 

  48.                  "shelfnet50_3343": ShelfNet503343,
    49. 

  49.                  "shelfnet50": ShelfNet50,
    50. 

  50.                  "shelfnet101": ShelfNet101,
    51. 

  51.                  "shufflenet_v2_x0_5": ShufflenetV2_x0_5,
    52. 

  52.                  "shufflenet_v2_x1_0": ShufflenetV2_x1_0,
    53. 

  53.                  "shufflenet_v2_x1_5": ShufflenetV2_x1_5,
    54. 

  54.                  "shufflenet_v2_x2_0": ShufflenetV2_x2_0,
    55. 

  55.                  "shufflenet_v2_custom5": CustomizedShuffleNetV2,
    56. 

  56.                  'darknet53': Darknet53,
    57. 

  57.                  'csp_darknet53': CSPDarknet53,
    58. 

  58.                  "resnext50": ResNeXt50,
    59. 

  59.                  "resnext101": ResNeXt101,
    60. 

  60.                  "googlenet_v1": googlenet_v1,
    61. 

  61.                  "efficientnet_b0": efficientnet.b0,
    62. 

  62.                  "efficientnet_b1": efficientnet.b1,
    63. 

  63.                  "efficientnet_b2": efficientnet.b2,
    64. 

  64.                  "efficientnet_b3": efficientnet.b3,
    65. 

  65.                  "efficientnet_b4": efficientnet.b4,
    66. 

  66.                  "efficientnet_b5": efficientnet.b5,
    67. 

  67.                  "efficientnet_b6": efficientnet.b6,
    68. 

  68.                  "efficientnet_b7": efficientnet.b7,
    69. 

  69.                  "efficientnet_b8": efficientnet.b8,
    70. 

  70.                  "efficientnet_l2": efficientnet.l2,
    71. 

  71.                  "CustomizedEfficientnet": efficientnet.CustomizedEfficientnet,
    72. 

  72.                  "regnetY200": regnet.RegNetY200,
    73. 

  73.                  "regnetY400": regnet.RegNetY400,
    74. 

  74.                  "regnetY600": regnet.RegNetY600,
    75. 

  75.                  "regnetY800": regnet.RegNetY800,
    76. 

  76.                  "custom_regnet": regnet.CustomRegNet,
    77. 

  77.                  "nas_regnet": regnet.NASRegNet,
    78. 

  78.                  "yolo_v5n": YoLoV5N,
    79. 

  79.                  "yolo_v5s": YoLoV5S,
    80. 

  80.                  "yolo_v5m": YoLoV5M,
    81. 

  81.                  "yolo_v5l": YoLoV5L,
    82. 

  82.                  "yolo_v5x": YoLoV5X,
    83. 

  83.                  "custom_yolov5": Custom_YoLoV5,
    84. 

  84.                  "ssd_mobilenet_v1": SSDMobileNetV1,
    85. 

  85.                  "ssd_lite_mobilenet_v2": SSDLiteMobileNetV2,
    86. 

  86.                  "repvgg_a0": repvgg.RepVggA0,
    87. 

  87.                  "repvgg_a1": repvgg.RepVggA1,
    88. 

  88.                  "repvgg_a2": repvgg.RepVggA2,
    89. 

  89.                  "repvgg_b0": repvgg.RepVggB0,
    90. 

  90.                  "repvgg_b1": repvgg.RepVggB1,
    91. 

  91.                  "repvgg_b2": repvgg.RepVggB2,
    92. 

  92.                  "repvgg_b3": repvgg.RepVggB3,
    93. 

  93.                  "repvgg_d2se": repvgg.RepVggD2SE,
    94. 

  94.                  "repvgg_custom": repvgg.RepVggCustom,
    95. 

  95.                  "ddrnet_23": DDRNet23,
    96. 

  96.                  "ddrnet_23_slim": DDRNet23Slim,
    97. 

  97.                  "custom_ddrnet_23": AnyBackBoneDDRNet23,
    98. 

  98.                  }
    99.
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