Skip to content

Commit 409b820

Browse files
committed
Fix weight loading for mobilenetv3
1 parent 7752969 commit 409b820

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
@@ -72,7 +72,7 @@ def get_encoder(name, in_channels=3, depth=5, weights=None, output_stride=32, **
7272
# convert timm- models to tu- models
7373
if is_equivalent_to_timm_universal(name):
7474
name = name.replace("timm-", "tu-")
75-
if "minimal" in name:
75+
if "mobilenetv3" in name:
7676
name = name.replace("tu-", "tu-tf_")
7777

7878
if name.startswith("tu-"):

0 commit comments

Comments
 (0)