iwslt params

wb14123 · Aug 30, 2017 · bd280ce · bd280ce
1 parent 4af5945
commit bd280ce
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/iwslt.py b/iwslt.py
@@ -2,12 +2,12 @@
 from model import Model
 
 m = Model(
-        '/data/dl-data/iwslt-2015/train.vi',
-        '/data/dl-data/iwslt-2015/train.en',
-        '/data/dl-data/iwslt-2015/tst2013.vi',
-        '/data/dl-data/iwslt-2015/tst2013.en',
-        '/data/dl-data/iwslt-2015/vocabs',
+        '/data/dl-data/iwslt15-google/train.en',
+        '/data/dl-data/iwslt15-google/train.vi',
+        '/data/dl-data/iwslt15-google/tst2013.en',
+        '/data/dl-data/iwslt15-google/tst2013.vi',
+        '/data/dl-data/iwslt15-google/vocab.en.bk',
         num_units=512, layers=2, dropout=0.2,
-        batch_size=64, learning_rate=0.001, output_dir='./output_iwslt')
+        batch_size=128, learning_rate=0.001, output_dir='./output_iwslt')
 
 m.train(5000000)