mirrors/pytorch-image-models
1
0
Fork 0
mirror of https://github.com/huggingface/pytorch-image-models.git synced 2025-06-03 15:01:08 +08:00
Code Issues Projects Releases Wiki Activity

Missed stashing of out_indices in model

Browse Source
This commit is contained in:
Ross Wightman 2019-11-05 09:55:55 -08:00
parent 35e8f0c5e7
commit 2393708650
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
timm/models/gen_efficientnet.py
View File

@ -1106,6 +1106,7 @@ class GenEfficientNetFeatures(_GenEfficientNet):
num_stages = max(out_indices) + 1 # FIXME reduce num stages created if not needed
#assert len(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
super(GenEfficientNetFeatures, self).__init__(