mirror of https://github.com/hpcaitech/ColossalAI
![]() * Add RoBERTa for RLHF Stage 2 & 3 (test) RoBERTa for RLHF Stage 2 & 3 (still in testing) * Revert "Add RoBERTa for RLHF Stage 2 & 3 (test)" This reverts commit |
||
---|---|---|
.. | ||
__init__.py | ||
prompt_dataset.py | ||
reward_dataset.py | ||
sft_dataset.py | ||
utils.py |