jhj0517
Add path
71b7d31
raw
history blame
395 Bytes
import functools
import os
PROJECT_ROOT_DIR = os.path.join(os.path.abspath(os.path.dirname(__file__)), "..", "..")
MODELS_DIR = os.path.join(PROJECT_ROOT_DIR, "models")
OUTPUTS_DIR = os.path.join(PROJECT_ROOT_DIR, "outputs")
@functools.lru_cache
def init_dirs():
for dir_path in [
MODELS_DIR,
OUTPUTS_DIR
]:
os.makedirs(dir_path, exist_ok=True)
init_dirs()