一个好玩的中文AI对联模型 输入格式 对联：您的上联 ，比如 对联：北国风光，千里冰封，万里雪飘 如果你想尝试 如果自己有GPU环境，可以参考我放在huggingface的示例代码 或者使用Google colab可以用这个简单的colab notebook 训练代码请参考我的github链接 如果想了解一些背景和讨论，可以看我的slides 架构 预训练使用 澜舟科技的孟子 T5 数据来源 对联数据集 https://github.com/wb14123/couplet dataset 标准输入输出seq2seq，T5使用 对联： 前缀，长度限制32字符 语言支持 默认简体中文 支持繁体中文，参考下面代码标记 is input traditional chinese=True 训练 我是用 Google Colab Pro（推荐，16G的GPU一个月才9.99！） 运行代码示例 简体中文示例 繁体中文
We use cookies for essential functionality and analytics. You can accept or reject analytics cookies.Cookie policy