Skip to content

Commit b53525e

Browse files
committed
Fix deprecation message
1 parent 1f422a1 commit b53525e

File tree

3 files changed

+3
-3
lines changed

3 files changed

+3
-3
lines changed

segmentation_models_pytorch/decoders/linknet/model.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -91,7 +91,7 @@ def __init__(
9191
decoder_use_batchnorm = kwargs.pop("decoder_use_batchnorm", None)
9292
if decoder_use_batchnorm is not None:
9393
warnings.warn(
94-
"The usage of decoder_use_batchnorm is deprecated. Please modify your code for use_norm",
94+
"The usage of decoder_use_batchnorm is deprecated. Please modify your code for decoder_use_norm",
9595
DeprecationWarning,
9696
stacklevel=2,
9797
)

segmentation_models_pytorch/decoders/manet/model.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -91,7 +91,7 @@ def __init__(
9191
decoder_use_batchnorm = kwargs.pop("decoder_use_batchnorm", None)
9292
if decoder_use_batchnorm is not None:
9393
warnings.warn(
94-
"The usage of decoder_use_batchnorm is deprecated. Please modify your code for use_norm",
94+
"The usage of decoder_use_batchnorm is deprecated. Please modify your code for decoder_use_norm",
9595
DeprecationWarning,
9696
stacklevel=2,
9797
)

segmentation_models_pytorch/decoders/unetplusplus/model.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -97,7 +97,7 @@ def __init__(
9797
decoder_use_batchnorm = kwargs.pop("decoder_use_batchnorm", None)
9898
if decoder_use_batchnorm is not None:
9999
warnings.warn(
100-
"The usage of decoder_use_batchnorm is deprecated. Please modify your code for use_norm",
100+
"The usage of decoder_use_batchnorm is deprecated. Please modify your code for decoder_use_norm",
101101
DeprecationWarning,
102102
stacklevel=2,
103103
)

0 commit comments

Comments
 (0)