Update train.py

PaddlePaddle · Jun 27, 2024 · 2bfc963 · 2bfc963
1 parent c217d57
commit 2bfc963
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tools/train.py b/tools/train.py
@@ -84,7 +84,7 @@ def main(args):
     split = dataset.get('split', None)
     do_eval = cfg.dic.get('do_eval', True)
     sampling_stride = cfg.dic.get('sampling_stride', 1)
-    cfg_output_dir = cfg.dic.get('output_dir', 'output')
+    cfg_output_dir = args.save_dir
     logger.info(cfg.__dict__)
 
     if not os.path.exists(cfg_output_dir):