From 6fdc9d8243098ef8f76990614c5fbeee6d75f122 Mon Sep 17 00:00:00 2001 From: dong-hyun Date: Fri, 2 Aug 2024 15:14:40 +0900 Subject: [PATCH] update code for ci pass --- timm/models/rdnet.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/timm/models/rdnet.py b/timm/models/rdnet.py index 7c2e2e27..162f80d5 100644 --- a/timm/models/rdnet.py +++ b/timm/models/rdnet.py @@ -247,7 +247,7 @@ class RDNet(nn.Module): ] dense_stages.append(nn.Sequential(*dense_stage_layers)) self.dense_stages = nn.Sequential(*dense_stages) - self.num_features = num_features + self.num_features = self.head_hidden_size = num_features # if head_norm_first == true, norm -> global pool -> fc ordering, like most other nets # otherwise pool -> norm -> fc, the default RDNet ordering (pretrained NV weights)