Skip to content
Snippets Groups Projects
Commit 683c077f authored by Low, Low Jian He (PG/R - Comp Sci & Elec Eng)'s avatar Low, Low Jian He (PG/R - Comp Sci & Elec Eng)
Browse files

adding dynamic path switching

parent 2fa36fb1
No related branches found
No related tags found
No related merge requests found
...@@ -4,7 +4,10 @@ from yacs.config import CfgNode as CN ...@@ -4,7 +4,10 @@ from yacs.config import CfgNode as CN
# CACHE_DIR_HAMER = "./_DATA" # CACHE_DIR_HAMER = "./_DATA"
# CACHE_DIR_HAMER = "/mnt/fast/nobackup/users/jl02958/hamer/_DATA" # CACHE_DIR_HAMER = "/mnt/fast/nobackup/users/jl02958/hamer/_DATA"
CACHE_DIR_HAMER = "/vol/research/SignHamerFeatures/hamer/_DATA" if os.path.exists("/vol/research/SignHamerFeatures/"):
CACHE_DIR_HAMER = "/vol/research/SignHamerFeatures/hamer/_DATA"
else:
CACHE_DIR_HAMER = "/mnt/fast/nobackup/users/jl02958/hamer/_DATA"
def to_lower(x: Dict) -> Dict: def to_lower(x: Dict) -> Dict:
""" """
......
...@@ -13,7 +13,13 @@ os.environ["PYOPENGL_PLATFORM"] = "egl" ...@@ -13,7 +13,13 @@ os.environ["PYOPENGL_PLATFORM"] = "egl"
# project root directory # project root directory
# ROOT_DIR = "/mnt/fast/nobackup/users/jl02958/hamer" # ROOT_DIR = "/mnt/fast/nobackup/users/jl02958/hamer"
ROOT_DIR = "/vol/research/SignHamerFeatures/hamer"
if os.path.exists("/vol/research/SignHamerFeatures/"):
ROOT_DIR = "/vol/research/SignHamerFeatures/hamer"
else:
ROOT_DIR = "/mnt/fast/nobackup/users/jl02958/hamer"
VIT_DIR = os.path.join(ROOT_DIR, "third-party/ViTPose") VIT_DIR = os.path.join(ROOT_DIR, "third-party/ViTPose")
class ViTPoseModel(object): class ViTPoseModel(object):
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment