Skip to content

Commit

Permalink
fix args parser for train_ms (#378)
Browse files Browse the repository at this point in the history
fix the bug for 81e034b
  • Loading branch information
tanyuzhou authored Jun 26, 2024
1 parent 669ac0e commit 7d0aa66
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion train_ms.py
Original file line number Diff line number Diff line change
Expand Up @@ -99,7 +99,7 @@ def run():
help="数据集文件夹路径,请注意,数据不再默认放在/logs文件夹下。如果需要用命令行配置,请声明相对于根目录的路径",
default=config.dataset_path,
)
args = parser.parse_known_args()
args, _ = parser.parse_known_args()
model_dir = os.path.join(args.model, config.train_ms_config.model)
if not os.path.exists(model_dir):
os.makedirs(model_dir, exist_ok=True)
Expand Down

0 comments on commit 7d0aa66

Please sign in to comment.