patch_megatron / data_process.sh
xingzhaohu's picture
Add files using upload-large-folder tool
add5224 verified
Raw
History Blame Contribute Delete
514 Bytes
python 1_extract_video_caption_from_npz.py
python 2_build_conversations.py
python 3_extract_frames.py --dataset-root="/data/hongwei/xzh_workspace/patch_megatron_code_new/my_datasets" --time-interval=1.0 --output-json="ted_dataset_extract_frame.json" --video-token="<video>"
python 4_convert_my_dataset_to_wds.py --dataset-root="/data/hongwei/xzh_workspace/patch_megatron_code_new/my_datasets" --json="ted_dataset_extract_frame.json" --max-samples-per-tar=10000 --train-split=9 --val-split=1 --test-split=0