seq2seq:这是带有Tensorflow 2的seq2seq模型结构 源码
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
用户评论