Shakespear GPT2

🚀 输出模型

该模型是在莎士比亚戏剧数据集上对 gpt2 进行微调后的版本，可用于生成类似莎士比亚风格的文本，具有一定的文学创作辅助价值。

🚀 快速开始

本部分暂未提供相关内容。

✨ 主要特性

• 基于原始的 gpt-2 模型，在自定义数据集上进行微调。
• 能够生成类似莎士比亚风格的文本，可能会重现戏剧的排版结构。

📚 详细文档

模型描述

该模型是在自定义数据集上对原始的 gpt-2 模型进行微调得到的。

预期用途和限制

此模型可用于生成类似莎士比亚风格的文本。需要注意的是，由于训练数据来自戏剧，生成的文本可能会重现这种排版结构。

训练和评估数据

使用莎士比亚戏剧语料库进行训练。

训练过程

训练超参数

训练过程中使用了以下超参数：

• 学习率（learning_rate）：5e-05
• 训练批次大小（train_batch_size）：1
• 评估批次大小（eval_batch_size）：8
• 随机种子（seed）：42
• 优化器（optimizer）：Adam，其中 betas=(0.9, 0.999)，epsilon=1e-08
• 学习率调度器类型（lr_scheduler_type）：线性
• 训练轮数（num_epochs）：2.0

框架版本

• Transformers 4.16.0.dev0
• Pytorch 1.10.0+cu111
• Datasets 1.18.0
• Tokenizers 0.11.0

📄 许可证

本模型采用 MIT 许可证。