From 50551c18606201f96664f35e0e5a84827cdaf453 Mon Sep 17 00:00:00 2001
From: Nick Pai <56616275+nick8592@users.noreply.github.com>
Date: Tue, 1 Aug 2023 21:29:33 +0800
Subject: [PATCH] Update predict_imagenet_label.py

to adapt latest version of `timm`
---
 predict_imagenet_label.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/predict_imagenet_label.py b/predict_imagenet_label.py
index 576f57d..79bf532 100644
--- a/predict_imagenet_label.py
+++ b/predict_imagenet_label.py
@@ -10,7 +10,8 @@
 import time
 import torch
 from timm.data import ImageDataset, create_loader, resolve_data_config
-from timm.models import apply_test_time_pool, create_model
+from timm.models import create_model
+from timm.models.layers import apply_test_time_pool
 from timm.utils import AverageMeter, setup_default_logging
 
 torch.backends.cudnn.benchmark = True