Commit 476b4a19 authored by yizhou-wang's avatar yizhou-wang
Browse files

Update prepare_data.py

parent 012ad57c
......@@ -202,7 +202,7 @@ def prepare_data(dataset, config_dict, data_dir, split, save_dir, viz=False, ove
if __name__ == "__main__":
args = parse_args()
data_root = args.data_root
if args.splits == '':
if args.split == '':
splits = None
else:
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