From 61e8414ad04d64429a29dac7fca1f51542b68c04 Mon Sep 17 00:00:00 2001 From: Fredo Guan Date: Thu, 12 Jan 2023 11:02:43 -0800 Subject: [PATCH] Update metaformers.py --- timm/models/metaformers.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/timm/models/metaformers.py b/timm/models/metaformers.py index ef43ded3..89651e02 100644 --- a/timm/models/metaformers.py +++ b/timm/models/metaformers.py @@ -710,7 +710,7 @@ class MetaFormer(nn.Module): if head_dropout > 0.0: self.head = self.head_fn(self.num_features, self.num_classes, head_dropout=self.head_dropout) else: - self.head = self.head_fn(self.num_featuers, self.num_classes) + self.head = self.head_fn(self.num_features, self.num_classes) self.apply(self._init_weights) @@ -737,7 +737,7 @@ class MetaFormer(nn.Module): if self.head_dropout > 0.0: self.head = self.head_fn(self.num_features, self.num_classes, head_dropout=self.head_dropout) else: - self.head = self.head_fn(self.num_featuers, self.num_classes) + self.head = self.head_fn(self.num_features, self.num_classes) def forward_head(self, x, pre_logits: bool = False): if pre_logits: