seq2seq
这是Tensorflow 2的seq2seq模型结构。
有三种模型架构,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
1