diff options
| -rw-r--r-- | src/jlens_v2.py | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/src/jlens_v2.py b/src/jlens_v2.py index 1bacbc6..c9b7019 100644 --- a/src/jlens_v2.py +++ b/src/jlens_v2.py @@ -7,6 +7,9 @@ import torch.nn as nn import numpy as np import pickle import os +import os, sys +# jlens_v2.py is in src/, model.py is in repo root +sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__)))) from collections import defaultdict def load_model(checkpoint_path, device='cuda'): |