Skip to content

GitLab

"docs/git@developer.sourcefind.cn:OpenDAS/deepspeed.git" did not exist on "b1ddea7fd94c52a7be76cec721d32d438681af83"
Commit 476b4a19 authored by yizhou-wang's avatar yizhou-wang
Browse files

Update prepare_data.py

parent 012ad57c
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tools/prepare_dataset/prepare_data.py
View file @ 476b4a19
...@@ -202,7 +202,7 @@ def prepare_data(dataset, config_dict, data_dir, split, save_dir, viz=False, ove ...@@ -202,7 +202,7 @@ def prepare_data(dataset, config_dict, data_dir, split, save_dir, viz=False, ove
if __name__ == "__main__": if __name__ == "__main__":
args = parse_args() args = parse_args()
data_root = args.data_root data_root = args.data_root
if args.splits == '': if args.split == '':
splits = None splits = None
else: else:
splits = args.split.split(',') splits = args.split.split(',')
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment