建议张量流seq2seq' lite'模型/数据集

时间:2016-01-16 18:12:46

标签: tensorflow

我想运行tensorflow seq2seq教程但不想使用大量内存。我现在不关心性能 - 我对调试更感兴趣。建议?

对于背景,我正在使用seq2seq教程here(并希望进行修改):

1 个答案:

答案 0 :(得分:0)

使用更小的输入文件,词汇大小,嵌入大小和/或层数来运行它。

尝试在[your_data_directory]中使用较小的培训文件......

> head -10000 giga-fren.release2.fr > giga-fren.release2.fr
> head -10000 giga-fren.release2.en > giga-fren.release2.en

尝试较小的词汇大小,如1000 ......

python translate.py
--data_dir [your_data_directory] --train_dir [checkpoints_directory]
--en_vocab_size=1000 --fr_vocab_size=1000

尝试较小的嵌入尺寸和/或较少的图层

python translate.py
--data_dir [your_data_directory] --train_dir [checkpoints_directory]
--size=128 --num_layers=1 --steps_per_checkpoint=50

该模型表现不佳,但运行速度更快,占地面积更小。