We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent eb3129b commit ec80704Copy full SHA for ec80704
segmentation_models_pytorch/encoders/__init__.py
@@ -46,7 +46,7 @@
46
def get_encoder(name, in_channels=3, depth=5, weights=None, output_stride=32, **kwargs):
47
48
if name.startswith("tu-"):
49
- name = name.lstrip("tu-")
+ name = name[3:]
50
encoder = TimmUniversalEncoder(
51
name=name,
52
in_channels=in_channels,
0 commit comments