feat(doc): add more info on train_on_split (#855)

2023-11-15 23:42:26 +09:00
parent 614cff4107
commit 306fe19c54
1 changed files with 1 additions and 0 deletions
--- a/README.md
+++ b/README.md
@@ -491,6 +491,7 @@ datasets:
    data_files: # Optional[str] path to source data files
    shards: # Optional[int] number of shards to split data into
    name: # Optional[str] name of dataset configuration to load
+    train_on_split: train # Optional[str] name of dataset split to load from

    # Optional[str] fastchat conversation type, only used with type: sharegpt
    conversation:  # Options (see Conversation 'name'): https://github.com/lm-sys/FastChat/blob/main/fastchat/conversation.py