Skip to content

Commit a4790b3

Browse files
committed
Remove **dd from two inception reset_classifier calls
1 parent e4e62f8 commit a4790b3

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

timm/models/inception_resnet_v2.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -323,7 +323,7 @@ def get_classifier(self) -> nn.Module:
323323

324324
def reset_classifier(self, num_classes: int, global_pool: str = 'avg'):
325325
self.num_classes = num_classes
326-
self.global_pool, self.classif = create_classifier(self.num_features, self.num_classes, pool_type=global_pool, **dd)
326+
self.global_pool, self.classif = create_classifier(self.num_features, self.num_classes, pool_type=global_pool)
327327

328328
def forward_features(self, x):
329329
x = self.conv2d_1a(x)

timm/models/inception_v4.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -341,7 +341,7 @@ def get_classifier(self) -> nn.Module:
341341
def reset_classifier(self, num_classes: int, global_pool: str = 'avg'):
342342
self.num_classes = num_classes
343343
self.global_pool, self.last_linear = create_classifier(
344-
self.num_features, self.num_classes, pool_type=global_pool, **dd)
344+
self.num_features, self.num_classes, pool_type=global_pool)
345345

346346
def forward_intermediates(
347347
self,

0 commit comments

Comments
 (0)