Migrations
/
pytorch-image-models
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Missed stashing of out_indices in model

Browse Source
pull/53/head
Ross Wightman 5 years ago
parent 35e8f0c5e7
commit 2393708650
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File

1
timm/models/gen_efficientnet.py
Unescape View File

@ -1106,6 +1106,7 @@ class GenEfficientNetFeatures(_GenEfficientNet):
num_stages = max(out_indices) + 1 # FIXME reduce num stages created if not needed num_stages = max(out_indices) + 1 # FIXME reduce num stages created if not needed
#assert len(block_args) >= num_stages - 1 #assert len(block_args) >= num_stages - 1
#block_args = block_args[:num_stages - 1] #block_args = block_args[:num_stages - 1]
self.out_indices = out_indices
# FIXME it would be nice if Python made this nicer without using kwargs and erasing IDE hints, etc # FIXME it would be nice if Python made this nicer without using kwargs and erasing IDE hints, etc
super(GenEfficientNetFeatures, self).__init__( super(GenEfficientNetFeatures, self).__init__(

Write Preview
Loading…
Cancel
Save