
Seq2Seq:基于TensorFlow 2的模型结构详解
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
本文章详细介绍基于TensorFlow 2的Seq2Seq模型结构,内容涵盖Seq2Seq基础原理、编码器-解码器架构及其在序列生成任务中的应用。
seq2seq是TensorFlow 2中的序列到序列模型结构。它包括三种架构:RNNSeq2Seq、RNNSeq2SeqWithAttention 和 TransformerSeq2Seq。该存储库包含了训练、评估、推断以及转换为保存的模型格式所需的脚本。
要开始训练,您可以运行以下命令:
```
python -m scripts.train \
--dataset-path data/*.txt \
--batch-size 2048 --dev-batch-size 2048 \
--epoch 90 --steps-per-epoch 250 --auto-encoding \
--learning-rate 2e-4 \
--device gpu \
--tensorboard-update-freq 50 --model-name Transformer
```
全部评论 (0)
还没有任何评论哟~


