push

cld2005 · Nov 28, 2017 · 50fee92 · 50fee92
1 parent 6562787
commit 50fee92
Show file tree

Hide file tree

Showing 3 changed files with 23 additions and 22 deletions.
diff --git a/.idea/workspace.xml b/.idea/workspace.xml
diff --git a/lstm_seq2seq_word_embed.py b/lstm_seq2seq_word_embed.py
@@ -330,7 +330,7 @@ def test_with_unseen_data():
                   epochs=epochs,
                   validation_split=0.2,verbose='1')
     # Save model
-    print ('saving....:models/s2s_south_park_epoch_%d.h5' % (i * 10 + 10))
-    model.save('models/s2s_south_park_epoch_%d.h5' % (i * 10 + 10))
+    print ('saving....:models/s2s_south_park_epoch_%d.h5' % (i * epochs + epochs))
+    model.save('models/s2s_south_park_epoch_%d.h5' % (i * epochs + epochs))
 
     test_with_unseen_data()
diff --git a/setup.sh b/setup.sh
@@ -4,4 +4,4 @@ mkdir glove.6B
 cd glove.6B
 wget https://s3.amazonaws.com/cld2005.glove.100/glove.6B.100d.txt
 cd ../LSTM_seq2seq
-git check out single_word
+git checkout single_word