Update preprocess_spec.py

332plim · Jun 6, 2023 · 13dee4b · 13dee4b
1 parent 576d12f
commit 13dee4b
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/prepare/preprocess_spec.py b/prepare/preprocess_spec.py
@@ -1,4 +1,8 @@
 import os
+import sys
+current_dir = os.path.dirname(os.path.abspath(__file__))
+parent_dir = os.path.abspath(os.path.join(current_dir, os.pardir))
+sys.path.append(parent_dir)
 import torch
 import argparse