python -u data_provider/s2generator_main.py  --root_path ./data/datasets/pretrain_data/ --start_seed 0  --end_seed 10  --max_input_dim 6  --max_output_dim 6  --length 768  --num_threads 4