ansh4749/TinyStories

---
license: cdla-sharing-1.0
task_categories:
- text-generation
language:
- en
---

Dataset containing synthetically generated (by GPT-3.5 and GPT-4) short stories that only use a small vocabulary.

Described in the following paper: https://arxiv.org/abs/2305.07759.

The models referred to in the paper were trained on TinyStories-train.txt (the file tinystories-valid.txt can be used for validation loss). These models can be found on Huggingface, at roneneldan/TinyStories-1M/3M/8M/28M/33M/1Layer-21M.

Additional resources:
tinystories_all_data.tar.gz - contains a superset of the stories together with metadata and the prompt that was used to create each story.

TinyStoriesV2-GPT4-train.txt - Is a new version of the dataset that is based on generations by GPT-4 only (the original dataset also has generations by GPT-3.5 which are of lesser quality). It contains all the examples in TinyStories.txt which were GPT-4 generated as a subset (but is significantly larger).

Evaluation_prompts.yaml: List of prompts used to evaluate our models (see paper)

许可证：CDLA-Sharing-1.0
任务类别：文本生成
语言：英语

本数据集包含由GPT-3.5（Generative Pre-trained Transformer 3.5）与GPT-4（Generative Pre-trained Transformer 4）合成生成的短篇故事，所有故事均采用精简词汇创作。
相关研究论文详见：https://arxiv.org/abs/2305.07759。

论文中提及的模型均基于TinyStories-train.txt进行训练（可使用tinystories-valid.txt文件计算验证集损失）。上述模型可在Huggingface平台的roneneldan/TinyStories-1M/3M/8M/28M/33M/1Layer-21M路径下获取。

附加资源如下：
1. tinystories_all_data.tar.gz：包含该数据集故事的超集，附带元数据及每篇故事对应的生成提示词。
2. TinyStoriesV2-GPT4-train.txt：本数据集的新版本，仅基于GPT-4生成的故事构建（原始数据集同时包含GPT-3.5生成的内容，质量相对较低）。该数据集以原始数据集中GPT-4生成的故事为子集，但整体规模显著更大。
3. Evaluation_prompts.yaml：用于评估模型的提示词列表（详见研究论文）。