fix(exp): change output_dir in base_exp and fix bug of exp mapping (#18)

Megvii-BaseDetection · Jul 20, 2021 · 5fe791e · 5fe791e
1 parent 69278c1
commit 5fe791e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -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",