github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-19 16:45:45 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
grpo-latest-dev-reward-update
ColossalAI/applications/ColossalChat/coati/distributed
History
YeAnbang d06042b434 rewrite reward fn
2025-05-01 11:28:05 +08:00
..
reward
rewrite reward fn
2025-05-01 11:28:05 +08:00
__init__.py
…
comm.py
…
consumer.py
rewrite reward fn
2025-05-01 11:28:05 +08:00
grpo_consumer.py
Support evaluation during training
2025-04-30 18:31:49 +08:00
inference_backend.py
Support evaluation during training
2025-04-30 18:31:49 +08:00
launch.py
fix bug
2025-04-30 22:53:12 +08:00
loss.py
[feat] Support DAPO (#6263)
2025-04-25 17:39:17 +08:00
producer.py
fix bug
2025-04-30 22:53:12 +08:00
README.md
…
utils.py
Support evaluation during training
2025-04-30 18:31:49 +08:00

README.md

Requirements

pip install cupy-cuda12x
python -m cupyx.tools.install_library --cuda 12.x --library nccl