diff --git a/paddlex/command.py b/paddlex/command.py
index 74079b9b708073c264f1baaacbd4f4fe075bbfe1..30be09842a47326aaaf1cde7e5f1006a6a24d33b 100644
--- a/paddlex/command.py
+++ b/paddlex/command.py
@@ -188,7 +188,7 @@ def main():
         val_value = float(args.val_value)
         test_value = float(args.test_value
                            if args.test_value is not None else 0)
-        save_dir = args.save_dir if args.save_dir is not None else "./"
+        save_dir = dataset_dir
 
         if not dataset_form in ["coco", "imagenet", "voc", "seg"]:
             logging.error(
@@ -198,8 +198,7 @@ def main():
             logging.error("The path of dataset to be splited doesn't exist.")
         if val_value <= 0 or val_value >= 1 or test_value < 0 or test_value >= 1 or val_value + test_value >= 1:
             logging.error("The value of split is not correct.")
-        if not osp.exists(save_dir):
-            logging.error("The path of saved split information doesn't exist.")
+
         pdx.tools.split.dataset_split(dataset_dir, dataset_form, val_value,
                                       test_value, save_dir)
 
diff --git a/paddlex/tools/split.py b/paddlex/tools/split.py
index ac6b7f01ed32140a1a1a57f2f2fe525b34a56ed8..f68b49364c53e0fb776ca811b7267117f035e938 100644
--- a/paddlex/tools/split.py
+++ b/paddlex/tools/split.py
@@ -21,7 +21,6 @@ from .dataset_split.seg_split import split_seg_dataset
 
 
 def dataset_split(dataset_dir, dataset_form, val_value, test_value, save_dir):
-    print(dataset_dir, dataset_form, val_value, test_value, save_dir)
     if dataset_form == "coco":
         train_num, val_num, test_num = split_coco_dataset(
             dataset_dir, val_value, test_value, save_dir)