diff --git a/config.py b/config.py
index 75526a3..c6a4d90 100644
--- a/config.py
+++ b/config.py
@@ -121,14 +121,14 @@ class Train_ms_config:
         config_path: str,
         env: Dict[str, any],
         # base: Dict[str, any],
-        model: str,
+        model_dir: str,
         num_workers: int,
         spec_cache: bool,
         keep_ckpts: int,
     ):
         self.env = env  # 需要加载的环境变量
         # self.base = base  # 底模配置
-        self.model = model  # 训练模型存储目录，该路径为相对于dataset_path的路径，而非项目根目录
+        self.model_dir = model_dir  # 训练模型存储目录，该路径为相对于dataset_path的路径，而非项目根目录
         self.config_path = config_path  # 配置文件路径
         self.num_workers = num_workers  # worker数量
         self.spec_cache = spec_cache  # 是否启用spec缓存
diff --git a/train_ms.py b/train_ms.py
index f508f77..a1e72d7 100644
--- a/train_ms.py
+++ b/train_ms.py
@@ -108,7 +108,7 @@ def run():
         help="Skip saving default style config and mean vector.",
     )
     args = parser.parse_args()
-    model_dir = os.path.join(args.model, config.train_ms_config.model)
+    model_dir = os.path.join(args.model, config.train_ms_config.model_dir)
     if not os.path.exists(model_dir):
         os.makedirs(model_dir)
     hps = utils.get_hparams_from_file(args.config)