Merge pull request #276 from mio2333/main

Append sys path for import_module
This commit is contained in:
Kohya S
2023-03-10 21:43:32 +09:00
committed by GitHub

View File

@@ -135,6 +135,8 @@ def train(args):
gc.collect()
# prepare network
import sys
sys.path.append(os.path.dirname(__file__))
print("import network module:", args.network_module)
network_module = importlib.import_module(args.network_module)