Enable dedicated MMSeg full mmcv runtime
This commit is contained in:
@@ -20,6 +20,7 @@ class Settings:
|
||||
log_dir: Path
|
||||
weights_root: Path
|
||||
task_conda_env: str
|
||||
mmseg_conda_env: str
|
||||
backend_conda_env: str
|
||||
weight_mode: str
|
||||
enable_shell_tasks: bool
|
||||
@@ -46,6 +47,7 @@ def get_settings() -> Settings:
|
||||
log_dir=log_dir,
|
||||
weights_root=weights_root,
|
||||
task_conda_env=os.getenv("SEG_TASK_CONDA_ENV", "seg_smp"),
|
||||
mmseg_conda_env=os.getenv("SEG_MMSEG_CONDA_ENV", "seg_mmcv"),
|
||||
backend_conda_env=os.getenv("SEG_BACKEND_CONDA_ENV", "seg_smp"),
|
||||
weight_mode=os.getenv("SEG_WEIGHT_MODE", "copy"),
|
||||
enable_shell_tasks=os.getenv("SEG_ENABLE_SHELL_TASKS", "1") == "1",
|
||||
|
||||
Reference in New Issue
Block a user