Skip to content

Commit c58b7cf

Browse files
committed
Fix typo error & Update doc
1 parent 029c190 commit c58b7cf

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

docs/encoders_timm.rst

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -16,8 +16,8 @@ Total number of encoders: 761 (579+182)
1616

1717
To use following encoders you have to add prefix ``tu-``, e.g. ``tu-adv_inception_v3``
1818

19-
Traditional-Style Models
20-
~~~~~~~~~~~~~~~~~~~~~~~~~
19+
Traditional-Style
20+
~~~~~~~~~~~~~~~~~
2121

2222
These models typically produce feature maps at the following downsampling scales relative to the input resolution: 1/2, 1/4, 1/8, 1/16, and 1/32
2323

segmentation_models_pytorch/encoders/timm_universal.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -117,7 +117,7 @@ def __init__(
117117
else:
118118
if "dla" in name:
119119
# For 'dla' models, out_indices starts at 0 and matches the input size.
120-
kwargs["out_indices"] = tuple(range(1, depth + 1))
120+
common_kwargs["out_indices"] = tuple(range(1, depth + 1))
121121

122122
self.model = timm.create_model(
123123
name, **_merge_kwargs_no_duplicates(common_kwargs, kwargs)

0 commit comments

Comments
 (0)