seq2seq:这是带有Tensorflow 2的seq2seq模型结构

时间:2024-04-07 18:06:03
【文件属性】:

文件名称:seq2seq:这是带有Tensorflow 2的seq2seq模型结构

文件大小:389KB

文件格式:ZIP

更新时间:2024-04-07 18:06:03

nlp seq2seq tensorflow2 Python

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


【文件预览】:
seq2seq-master
----requirements-dev.txt(38B)
----pyproject.toml(135B)
----tests()
--------test_utils.py(499B)
--------test_data.py(1KB)
--------data()
--------__init__.py(0B)
--------test_layer.py(3KB)
--------test_search.py(1KB)
--------test_model.py(883B)
----resources()
--------sp-model()
--------configs()
----seq2seq()
--------layer.py(9KB)
--------utils.py(3KB)
--------search.py(8KB)
--------__init__.py(105B)
--------model.py(12KB)
--------data.py(2KB)
----requirements.txt(58B)
----setup.py(359B)
----.gitignore(298B)
----.circleci()
--------config.yml(2KB)
----README.md(10KB)
----scripts()
--------inference.py(4KB)
--------train.py(7KB)
--------convert_to_savedmodel.py(5KB)
--------evaluate.py(5KB)
----.vscode()
--------settings.json(202B)
--------extensions.json(62B)

网友评论