From 5fe791ed896197027d6d4a332098ff1c23c0afaa Mon Sep 17 00:00:00 2001 From: Feng Wang Date: Tue, 20 Jul 2021 17:16:11 +0800 Subject: [PATCH] fix(exp): change output_dir in base_exp and fix bug of exp mapping (#18) --- yolox/exp/base_exp.py | 2 +- yolox/exp/build.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/yolox/exp/base_exp.py b/yolox/exp/base_exp.py index 939833d2e..3e9307073 100644 --- a/yolox/exp/base_exp.py +++ b/yolox/exp/base_exp.py @@ -20,7 +20,7 @@ class BaseExp(metaclass=ABCMeta): def __init__(self): self.seed = None - self.output_dir = "/data/YOLOX_outputs" + self.output_dir = "./YOLOX_outputs" self.print_interval = 100 self.eval_interval = 10 diff --git a/yolox/exp/build.py b/yolox/exp/build.py index 55aa46751..7e1ba6395 100644 --- a/yolox/exp/build.py +++ b/yolox/exp/build.py @@ -22,7 +22,7 @@ def get_exp_by_name(exp_name): yolox_path = os.path.dirname(os.path.dirname(yolox.__file__)) filedict = { "yolox-s": "yolox_s.py", - "yolox-m": "yolox_l.py", + "yolox-m": "yolox_m.py", "yolox-l": "yolox_l.py", "yolox-x": "yolox_x.py", "yolox-tiny": "yolox_tiny.py",