Update train_nsf_sim_cache_sid_load_pretrain.py

2023-07-26 18:05:44 +08:00 · 2023-07-26 18:05:44 +08:00 · 23642ac22a
parent 2a71c31b66
commit 23642ac22a
1 changed files with 5 additions and 1 deletions
--- a/train_nsf_sim_cache_sid_load_pretrain.py
+++ b/train_nsf_sim_cache_sid_load_pretrain.py
@ -1,5 +1,9 @@
-import os
+import os,sys
+now_dir = os.getcwd()
+sys.path.append(os.path.join(now_dir))
+sys.path.append(os.path.join(now_dir, "train"))

+from lib.train import utils
 from lib.train import utils
 import datetime