-
StoryTellerAI的LeetCode刷题笔记-谷歌师兄:基于Pytorch实现的StoryTellarAI
资源介绍
谷歌师兄的leetcode刷题笔记说故事的人
文本生成神经网络,由格林兄弟的童话故事、刘易斯卡罗尔的爱丽丝梦游仙境和尼尔盖曼的
Coraline
训练。
强烈建议使用
GPU
连接进行染色,因为在
CPU
上训练可能需要
10
多个小时。
重要笔记请将
notebook
中的以下变量替换为存储库中的
.txt
文件,
因为这个笔记本链接到我自己的
Google
Drive。
fairtales_url
:
确保它链接到
FairytalesByTheBrothersGrimm.txt
alice_url
:
确保它链接到
AlicesAdvanturesInWonderland.txt
coraline_url
:
确保它链接到
Coraline.pdf
checkpoint_url
:
训练后保存模型检查点的任何位置这是一项正在进行的工作,将来我会优化这个网络以获得最大的准确性。