Refine Torch Load Model Path

magicleap#12
Visual-Feature-Matching · Jul 9, 2020 · 47c6723 · 47c6723
1 parent 82b22d0
commit 47c6723
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/models/superglue.py b/models/superglue.py
@@ -223,7 +223,7 @@ def __init__(self, config):
         assert self.config['weights'] in ['indoor', 'outdoor']
         path = Path(__file__).parent
         path = path / 'weights/superglue_{}.pth'.format(self.config['weights'])
-        self.load_state_dict(torch.load(path))
+        self.load_state_dict(torch.load(str(path)))
         print('Loaded SuperGlue model (\"{}\" weights)'.format(
             self.config['weights']))