Skip to content

Commit 2393708

Browse files
committed
Missed stashing of out_indices in model
1 parent 35e8f0c commit 2393708

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

timm/models/gen_efficientnet.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1106,6 +1106,7 @@ def __init__(self, block_args, out_indices=(0, 1, 2, 3, 4), feature_location='pr
11061106
num_stages = max(out_indices) + 1 # FIXME reduce num stages created if not needed
11071107
#assert len(block_args) >= num_stages - 1
11081108
#block_args = block_args[:num_stages - 1]
1109+
self.out_indices = out_indices
11091110

11101111
# FIXME it would be nice if Python made this nicer without using kwargs and erasing IDE hints, etc
11111112
super(GenEfficientNetFeatures, self).__init__(

0 commit comments

Comments
 (0)