LAION-AI / LAION-AI/Open-Assistant

DatasetEntry Roadmap

Open
#2,993 0 comments 0 reactions 1 assignee View on GitHub

@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

Open the contributing guide

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.