Skip to content

Commit ec80704

Browse files
lorenzo-parklorenzo@cricket
andauthored
Fix timm universal prefix replacement (#452)
Co-authored-by: lorenzo@cricket <lorenzopark@postech.ac.kr>
1 parent eb3129b commit ec80704

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

segmentation_models_pytorch/encoders/__init__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -46,7 +46,7 @@
4646
def get_encoder(name, in_channels=3, depth=5, weights=None, output_stride=32, **kwargs):
4747

4848
if name.startswith("tu-"):
49-
name = name.lstrip("tu-")
49+
name = name[3:]
5050
encoder = TimmUniversalEncoder(
5151
name=name,
5252
in_channels=in_channels,

0 commit comments

Comments
 (0)