├── ActivityNet └── activitynet_annotation │ ├── train.json │ └── val_2.json ├── Charades └── charades_annotation │ ├── train.json │ └── val.json ├── LICENSE.txt ├── README.md ├── TimeZero_TechReport.pdf ├── configs ├── ddp.yaml ├── qwen2vl_sft_config.yaml ├── zero2.yaml └── zero3.yaml ├── data_configs.py ├── dataset ├── activitynet │ ├── test.json │ └── val.json └── charades-sta │ ├── charades_test.json │ ├── charades_test_iid.json │ ├── charades_test_ood.json │ ├── novel_composition.json │ ├── novel_word.json │ └── test_trivial.json ├── environment.yml ├── evaluate.py ├── preprocess_dataset.py ├── scripts ├── evaluate.sh ├── preprocess_video.sh ├── run_grpo_video.sh └── zero3_offload.json └── src └── open_r1 ├── __init__.py ├── __pycache__ └── __init__.cpython-310.pyc ├── evaluate.py ├── generate.py ├── grpo.py ├── grpo_video.py ├── sft.py └── trainer ├── __init__.py ├── __pycache__ ├── __init__.cpython-310.pyc ├── grpo_trainer.cpython-310.pyc ├── grpo_trainer_video.cpython-310.pyc ├── grpo_trainer_video_git.cpython-310.pyc ├── grpo_trainer_video_test.cpython-310.pyc ├── vllm_grpo_trainer.cpython-310.pyc ├── vllm_grpo_trainer_video.cpython-310.pyc ├── vllm_grpo_trainer_video.cpython-310.pyc.139618446054256 ├── vllm_grpo_trainer_video.cpython-310.pyc.139812201141104 ├── vllm_grpo_trainer_video.cpython-310.pyc.140058536612720 ├── vllm_grpo_trainer_video.cpython-310.pyc.140202798732144 └── vllm_grpo_trainer_video_test.cpython-310.pyc ├── grpo_trainer.py ├── grpo_trainer_video.py ├── vllm_grpo_trainer.py └── vllm_grpo_trainer_video.py /ActivityNet/activitynet_annotation/train.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/www-Ye/Time-R1/HEAD/ActivityNet/activitynet_annotation/train.json -------------------------------------------------------------------------------- /ActivityNet/activitynet_annotation/val_2.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/www-Ye/Time-R1/HEAD/ActivityNet/activitynet_annotation/val_2.json -------------------------------------------------------------------------------- /Charades/charades_annotation/train.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/www-Ye/Time-R1/HEAD/Charades/charades_annotation/train.json -------------------------------------------------------------------------------- /Charades/charades_annotation/val.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/www-Ye/Time-R1/HEAD/Charades/charades_annotation/val.json -------------------------------------------------------------------------------- /LICENSE.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/www-Ye/Time-R1/HEAD/LICENSE.txt -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/www-Ye/Time-R1/HEAD/README.md -------------------------------------------------------------------------------- /TimeZero_TechReport.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/www-Ye/Time-R1/HEAD/TimeZero_TechReport.pdf -------------------------------------------------------------------------------- /configs/ddp.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/www-Ye/Time-R1/HEAD/configs/ddp.yaml -------------------------------------------------------------------------------- /configs/qwen2vl_sft_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/www-Ye/Time-R1/HEAD/configs/qwen2vl_sft_config.yaml -------------------------------------------------------------------------------- /configs/zero2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/www-Ye/Time-R1/HEAD/configs/zero2.yaml -------------------------------------------------------------------------------- /configs/zero3.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/www-Ye/Time-R1/HEAD/configs/zero3.yaml -------------------------------------------------------------------------------- /data_configs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/www-Ye/Time-R1/HEAD/data_configs.py -------------------------------------------------------------------------------- /dataset/activitynet/test.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/www-Ye/Time-R1/HEAD/dataset/activitynet/test.json -------------------------------------------------------------------------------- /dataset/activitynet/val.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/www-Ye/Time-R1/HEAD/dataset/activitynet/val.json -------------------------------------------------------------------------------- /dataset/charades-sta/charades_test.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/www-Ye/Time-R1/HEAD/dataset/charades-sta/charades_test.json -------------------------------------------------------------------------------- /dataset/charades-sta/charades_test_iid.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/www-Ye/Time-R1/HEAD/dataset/charades-sta/charades_test_iid.json -------------------------------------------------------------------------------- /dataset/charades-sta/charades_test_ood.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/www-Ye/Time-R1/HEAD/dataset/charades-sta/charades_test_ood.json -------------------------------------------------------------------------------- /dataset/charades-sta/novel_composition.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/www-Ye/Time-R1/HEAD/dataset/charades-sta/novel_composition.json -------------------------------------------------------------------------------- /dataset/charades-sta/novel_word.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/www-Ye/Time-R1/HEAD/dataset/charades-sta/novel_word.json -------------------------------------------------------------------------------- /dataset/charades-sta/test_trivial.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/www-Ye/Time-R1/HEAD/dataset/charades-sta/test_trivial.json -------------------------------------------------------------------------------- /environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/www-Ye/Time-R1/HEAD/environment.yml -------------------------------------------------------------------------------- /evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/www-Ye/Time-R1/HEAD/evaluate.py -------------------------------------------------------------------------------- /preprocess_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/www-Ye/Time-R1/HEAD/preprocess_dataset.py -------------------------------------------------------------------------------- /scripts/evaluate.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/www-Ye/Time-R1/HEAD/scripts/evaluate.sh -------------------------------------------------------------------------------- /scripts/preprocess_video.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/www-Ye/Time-R1/HEAD/scripts/preprocess_video.sh -------------------------------------------------------------------------------- /scripts/run_grpo_video.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/www-Ye/Time-R1/HEAD/scripts/run_grpo_video.sh -------------------------------------------------------------------------------- /scripts/zero3_offload.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/www-Ye/Time-R1/HEAD/scripts/zero3_offload.json -------------------------------------------------------------------------------- /src/open_r1/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/open_r1/__pycache__/__init__.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/www-Ye/Time-R1/HEAD/src/open_r1/__pycache__/__init__.cpython-310.pyc -------------------------------------------------------------------------------- /src/open_r1/evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/www-Ye/Time-R1/HEAD/src/open_r1/evaluate.py -------------------------------------------------------------------------------- /src/open_r1/generate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/www-Ye/Time-R1/HEAD/src/open_r1/generate.py -------------------------------------------------------------------------------- /src/open_r1/grpo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/www-Ye/Time-R1/HEAD/src/open_r1/grpo.py -------------------------------------------------------------------------------- /src/open_r1/grpo_video.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/www-Ye/Time-R1/HEAD/src/open_r1/grpo_video.py -------------------------------------------------------------------------------- /src/open_r1/sft.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/www-Ye/Time-R1/HEAD/src/open_r1/sft.py -------------------------------------------------------------------------------- /src/open_r1/trainer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/www-Ye/Time-R1/HEAD/src/open_r1/trainer/__init__.py -------------------------------------------------------------------------------- /src/open_r1/trainer/__pycache__/__init__.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/www-Ye/Time-R1/HEAD/src/open_r1/trainer/__pycache__/__init__.cpython-310.pyc -------------------------------------------------------------------------------- /src/open_r1/trainer/__pycache__/grpo_trainer.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/www-Ye/Time-R1/HEAD/src/open_r1/trainer/__pycache__/grpo_trainer.cpython-310.pyc -------------------------------------------------------------------------------- /src/open_r1/trainer/__pycache__/grpo_trainer_video.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/www-Ye/Time-R1/HEAD/src/open_r1/trainer/__pycache__/grpo_trainer_video.cpython-310.pyc -------------------------------------------------------------------------------- /src/open_r1/trainer/__pycache__/grpo_trainer_video_git.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/www-Ye/Time-R1/HEAD/src/open_r1/trainer/__pycache__/grpo_trainer_video_git.cpython-310.pyc -------------------------------------------------------------------------------- /src/open_r1/trainer/__pycache__/grpo_trainer_video_test.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/www-Ye/Time-R1/HEAD/src/open_r1/trainer/__pycache__/grpo_trainer_video_test.cpython-310.pyc -------------------------------------------------------------------------------- /src/open_r1/trainer/__pycache__/vllm_grpo_trainer.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/www-Ye/Time-R1/HEAD/src/open_r1/trainer/__pycache__/vllm_grpo_trainer.cpython-310.pyc -------------------------------------------------------------------------------- /src/open_r1/trainer/__pycache__/vllm_grpo_trainer_video.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/www-Ye/Time-R1/HEAD/src/open_r1/trainer/__pycache__/vllm_grpo_trainer_video.cpython-310.pyc -------------------------------------------------------------------------------- /src/open_r1/trainer/__pycache__/vllm_grpo_trainer_video.cpython-310.pyc.139618446054256: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/www-Ye/Time-R1/HEAD/src/open_r1/trainer/__pycache__/vllm_grpo_trainer_video.cpython-310.pyc.139618446054256 -------------------------------------------------------------------------------- /src/open_r1/trainer/__pycache__/vllm_grpo_trainer_video.cpython-310.pyc.139812201141104: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/www-Ye/Time-R1/HEAD/src/open_r1/trainer/__pycache__/vllm_grpo_trainer_video.cpython-310.pyc.139812201141104 -------------------------------------------------------------------------------- /src/open_r1/trainer/__pycache__/vllm_grpo_trainer_video.cpython-310.pyc.140058536612720: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/www-Ye/Time-R1/HEAD/src/open_r1/trainer/__pycache__/vllm_grpo_trainer_video.cpython-310.pyc.140058536612720 -------------------------------------------------------------------------------- /src/open_r1/trainer/__pycache__/vllm_grpo_trainer_video.cpython-310.pyc.140202798732144: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/www-Ye/Time-R1/HEAD/src/open_r1/trainer/__pycache__/vllm_grpo_trainer_video.cpython-310.pyc.140202798732144 -------------------------------------------------------------------------------- /src/open_r1/trainer/__pycache__/vllm_grpo_trainer_video_test.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/www-Ye/Time-R1/HEAD/src/open_r1/trainer/__pycache__/vllm_grpo_trainer_video_test.cpython-310.pyc -------------------------------------------------------------------------------- /src/open_r1/trainer/grpo_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/www-Ye/Time-R1/HEAD/src/open_r1/trainer/grpo_trainer.py -------------------------------------------------------------------------------- /src/open_r1/trainer/grpo_trainer_video.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/www-Ye/Time-R1/HEAD/src/open_r1/trainer/grpo_trainer_video.py -------------------------------------------------------------------------------- /src/open_r1/trainer/vllm_grpo_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/www-Ye/Time-R1/HEAD/src/open_r1/trainer/vllm_grpo_trainer.py -------------------------------------------------------------------------------- /src/open_r1/trainer/vllm_grpo_trainer_video.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/www-Ye/Time-R1/HEAD/src/open_r1/trainer/vllm_grpo_trainer_video.py --------------------------------------------------------------------------------