agentscope-ai / agentscope-ai/Trinity-RFT
[FEATURE] More granular config options for control of liger and optimizer
Đang mở
enhancement
- Ngôn ngữ chính
- Python
- Star
- 701
- Fork
- 79
- Merge trung bình
- 8 giờ 7 phút
- Pull request đã merge (30 ngày)
- 1
Mô tả
1. Would be nice to add `optimizer_fused`, `optimizer_foreach` config options. It allows to better control the memory footprint of the optimizer
Upstream issue:
- https://github.com/volcengine/verl/issues/2556
2. Would be nice to better control `_apply_liger_kernel_to_instance(model=actor_module)` ->
`_apply_liger_kernel_to_instance(model=actor_module, fused_linear_cross_entropy=liger_fused_linear_cross_entropy, cross_entropy=liger_cross_entropy)`. This is needed because otherwise Verl's fused FusedLinearCrossEntropy kernel conflicts with Liger kernels - at least used to conflict
Upstream issue:
- https://github.com/volcengine/verl/issues/2609
Hướng dẫn đóng góp
Đánh giá
Issue này chưa được đánh giá.