From b8b91acbd5688a2857acc46d4f1cb489ab38bf21 Mon Sep 17 00:00:00 2001 From: kaiyangzhou Date: Wed, 11 Aug 2021 16:50:56 +0800 Subject: [PATCH] fpath = osp.abspath(osp.expanduser(fpath)) --- torchreid/utils/torchtools.py | 1 + 1 file changed, 1 insertion(+) diff --git a/torchreid/utils/torchtools.py b/torchreid/utils/torchtools.py index b6df8d8..e854278 100644 --- a/torchreid/utils/torchtools.py +++ b/torchreid/utils/torchtools.py @@ -77,6 +77,7 @@ def load_checkpoint(fpath): """ if fpath is None: raise ValueError('File path is None') + fpath = osp.abspath(osp.expanduser(fpath)) if not osp.exists(fpath): raise FileNotFoundError('File is not found at "{}"'.format(fpath)) map_location = None if torch.cuda.is_available() else 'cpu'