From 88982d792c5a8d06fe02f2d73002c62a4bba0c8f Mon Sep 17 00:00:00 2001 From: Xiaoda Zhang Date: Mon, 1 Jun 2020 20:12:17 +0800 Subject: [PATCH] fix some codestyle warnings --- model_zoo/wide_and_deep/src/process_data.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/model_zoo/wide_and_deep/src/process_data.py b/model_zoo/wide_and_deep/src/process_data.py index 37b38b0bbbb..acf618297f8 100644 --- a/model_zoo/wide_and_deep/src/process_data.py +++ b/model_zoo/wide_and_deep/src/process_data.py @@ -248,8 +248,8 @@ def random_split_trans2h5(in_file_path, output_path, criteo_stats, part_rows=200 if __name__ == "__main__": parser = argparse.ArgumentParser(description="Get and Process datasets") - parser.add_argument("--raw_data_path", default="/opt/npu/data/origin_criteo_data/", help="The path to save dataset") - parser.add_argument("--output_path", default="/opt/npu/data/origin_criteo_data/h5_data/", + parser.add_argument("--raw_data_path", default="./raw_data", help="The path to save dataset") + parser.add_argument("--output_path", default="./output", help="The path to save dataset") args, _ = parser.parse_known_args() base_path = args.raw_data_path