├── LICENSE ├── README.md ├── requirements.txt ├── src ├── dataset │ ├── __init__.py │ └── dataset.py ├── test.py ├── train.py ├── trainer │ ├── __init__.py │ └── grpo_trainer.py └── utils │ ├── prepare_aqa.py │ ├── prepare_mmau.py │ ├── prepare_musicbench_key.py │ ├── prepare_musicbench_tempo.py │ ├── prepare_musicbench_time.py │ ├── rewards.py │ └── show_acc.py ├── test_mmau.sh └── train_omni_grpo.sh /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuaijiang/Ke-Omni-R/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuaijiang/Ke-Omni-R/HEAD/README.md -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | trl 2 | deepspeed 3 | math_verify 4 | tensorboard 5 | tabulate -------------------------------------------------------------------------------- /src/dataset/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/dataset/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuaijiang/Ke-Omni-R/HEAD/src/dataset/dataset.py -------------------------------------------------------------------------------- /src/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuaijiang/Ke-Omni-R/HEAD/src/test.py -------------------------------------------------------------------------------- /src/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuaijiang/Ke-Omni-R/HEAD/src/train.py -------------------------------------------------------------------------------- /src/trainer/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/trainer/grpo_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuaijiang/Ke-Omni-R/HEAD/src/trainer/grpo_trainer.py -------------------------------------------------------------------------------- /src/utils/prepare_aqa.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuaijiang/Ke-Omni-R/HEAD/src/utils/prepare_aqa.py -------------------------------------------------------------------------------- /src/utils/prepare_mmau.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuaijiang/Ke-Omni-R/HEAD/src/utils/prepare_mmau.py -------------------------------------------------------------------------------- /src/utils/prepare_musicbench_key.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuaijiang/Ke-Omni-R/HEAD/src/utils/prepare_musicbench_key.py -------------------------------------------------------------------------------- /src/utils/prepare_musicbench_tempo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuaijiang/Ke-Omni-R/HEAD/src/utils/prepare_musicbench_tempo.py -------------------------------------------------------------------------------- /src/utils/prepare_musicbench_time.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuaijiang/Ke-Omni-R/HEAD/src/utils/prepare_musicbench_time.py -------------------------------------------------------------------------------- /src/utils/rewards.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuaijiang/Ke-Omni-R/HEAD/src/utils/rewards.py -------------------------------------------------------------------------------- /src/utils/show_acc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuaijiang/Ke-Omni-R/HEAD/src/utils/show_acc.py -------------------------------------------------------------------------------- /test_mmau.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuaijiang/Ke-Omni-R/HEAD/test_mmau.sh -------------------------------------------------------------------------------- /train_omni_grpo.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shuaijiang/Ke-Omni-R/HEAD/train_omni_grpo.sh --------------------------------------------------------------------------------