From de86314655de1a14655297b66cc9f66e2fd181de Mon Sep 17 00:00:00 2001
From: contrastive <contrastive2021@163.com>
Date: Mon, 29 Mar 2021 08:23:34 +0800
Subject: [PATCH] Update TNT

---
 timm/models/tnt.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/timm/models/tnt.py b/timm/models/tnt.py
index e646b8bb..4d5b0723 100644
--- a/timm/models/tnt.py
+++ b/timm/models/tnt.py
@@ -24,7 +24,7 @@ def _cfg(url='', **kwargs):
         'num_classes': 1000, 'input_size': (3, 224, 224), 'pool_size': None,
         'crop_pct': .9, 'interpolation': 'bicubic',
         'mean': IMAGENET_DEFAULT_MEAN, 'std': IMAGENET_DEFAULT_STD,
-        'classifier': 'head',
+        'first_conv': 'pixel_embed.proj', 'classifier': 'head',
         **kwargs
     }