diff --git a/src/super_gradients/training/models/segmentation_models/stdc.py b/src/super_gradients/training/models/segmentation_models/stdc.py index f4516958c4..6c3a72a469 100644 --- a/src/super_gradients/training/models/segmentation_models/stdc.py +++ b/src/super_gradients/training/models/segmentation_models/stdc.py @@ -584,6 +584,7 @@ def _separate_lr_multiply_params(self): @register_model(Models.STDC_CUSTOM) +@register_model("custom_stdc") # deprecated naming convention. will be dropped in v4 class CustomSTDCSegmentation(STDCSegmentationBase): """ Fully customized STDC Segmentation factory module.