add couplet model

wb14123 · Oct 23, 2017 · 3106da8 · 3106da8
1 parent e5f61ca
commit 3106da8
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 7 deletions.
diff --git a/couplet.py b/couplet.py
@@ -8,6 +8,6 @@
         '/data/dl-data/couplet/test/out.txt',
         '/data/dl-data/couplet/vocabs',
         num_units=1024, layers=4, dropout=0.2,
-        batch_size=32, learning_rate=0.001, output_dir='./couplet')
-
-m.train(5000000)
+        batch_size=32, learning_rate=0.001,
+        output_dir='/data/dl-data/models/tf-lib/output_couplet',
+        restore_model=False)
diff --git a/nmt.py b/nmt.py
@@ -4,10 +4,12 @@
 m = Model(
         '/data/dl-data/wmt-2016/train.tok.clean.bpe.32000.de',
         '/data/dl-data/wmt-2016/train.tok.clean.bpe.32000.en',
-        '/data/dl-data/wmt-2016/newstest2016.tok.de',
-        '/data/dl-data/wmt-2016/newstest2016.tok.en',
+        '/data/dl-data/wmt-2016/newstest2016.tok.bpe.32000.de',
+        '/data/dl-data/wmt-2016/newstest2016.tok.bpe.32000.en',
         '/data/dl-data/wmt-2016/vocab.bpe.32000.bk',
         num_units=1024, layers=4, dropout=0.2,
-        batch_size=32, learning_rate=0.001, output_dir='./output_nmt')
+        batch_size=32, learning_rate=0.0001,
+        output_dir='/data/dl-data/models/tf-lib/output_nmt',
+        restore_model=True)
 
-m.train(5000000)
+m.train(5000000, start=102500)