SUSHANT283/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
任务类别：
- 文本生成（text-generation）
语言：
- 英语（en）

本数据集涵盖由GPT-3.5与GPT-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：用于评估模型的提示词列表（详见论文）。