LAION-AI / LAION-AI/Open-Assistant
DatasetEntry Roadmap
Open
@CloseChoice is already working on this.
Since May 1, 2023.
ml
- Dominant language
- Python
- Stars
- 37.4k
- Forks
- 3.3k
- PR merge metrics
- No merged PRs in 30d
Description
This is the suggestion for a major refactoring of the data preprocessing in the model_training trainer_sft, trainer_rm and trainer_rl logic. The crucial point here is 4. but we need the previous points as preparation.
- 1. make RM usable with DatasetEntry (also make sure that oasst dataset works)
- 2. refactor DatasetEntry to use different classes (DatasetEntryRM, DatasetEntryRL, DatasetEntrySFT)
- 3. generalize preprocessing and filtering (removing "as an AI language model", etc.) and apply this to all datasets (take the performance hit for now)
- 4. split dataset preprocessing/tokenization and training run in different steps
- 5. add testing properties for the preprocessed & tokenized data
Contributor guide
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Assessment
This issue has not been assessed yet.