We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent f438ddb commit 85e0a35Copy full SHA for 85e0a35
efficientnet_pytorch/model.py
@@ -261,12 +261,12 @@ def extract_endpoints(self, inputs):
261
drop_connect_rate *= float(idx) / len(self._blocks) # scale drop connect_rate
262
x = block(x, drop_connect_rate=drop_connect_rate)
263
if prev_x.size(2) > x.size(2):
264
- endpoints[f'reduction_{len(endpoints)+1}'] = prev_x
+ endpoints['reduction_{}'.format(len(endpoints)+1)] = prev_x
265
prev_x = x
266
267
# Head
268
x = self._swish(self._bn1(self._conv_head(x)))
269
- endpoints[f'reduction_{len(endpoints)+1}'] = x
+ endpoints['reduction_{}'.format(len(endpoints)+1)] = x
270
271
return endpoints
272
0 commit comments