From 27060edcfe1bf2ddb9d62c4bf328a0f87cce6324 Mon Sep 17 00:00:00 2001
From: jim4399266 <32267011+jim4399266@users.noreply.github.com>
Date: Tue, 16 Nov 2021 16:06:39 +0800
Subject: [PATCH] Update helpers.py

add logger info
---
 timm/models/helpers.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/timm/models/helpers.py b/timm/models/helpers.py
index 3404c47c..4bacde6f 100644
--- a/timm/models/helpers.py
+++ b/timm/models/helpers.py
@@ -186,6 +186,7 @@ def load_pretrained(model, default_cfg=None, num_classes=1000, in_chans=3, filte
         return
     if pth_file_path is not None:
         # load pretrained from local file
+        _logger.info(f'Loading pretrained weights from local file: ({pth_file_path})')
         state_dict = torch.load(pth_file_path, map_location='cpu')
     elif hf_hub_id and has_hf_hub(necessary=not pretrained_url):
         _logger.info(f'Loading pretrained weights from Hugging Face hub ({hf_hub_id})')