better path compatibility with notebooks

gear · Jan 28, 2020 · 81878f6 · 81878f6
1 parent e26e8ae
commit 81878f6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/utils.py b/utils.py
@@ -11,7 +11,7 @@
 from time import perf_counter
 from torch.utils import data
 
-dataf = os.path.expanduser("./data/")
+dataf = os.path.expanduser("{}/data/".format(os.path.dirname(__file__)))
 
 def parse_index_file(filename):
     """Parse index file."""