From 9f5b53486a28dd30e2dbf939c00ee925ec016b40 Mon Sep 17 00:00:00 2001 From: Joerg Franke Date: Mon, 25 Jun 2018 13:09:16 +0200 Subject: [PATCH] update mann --- adnc/model/mann.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/adnc/model/mann.py b/adnc/model/mann.py index 1db6c63..52f6ff0 100755 --- a/adnc/model/mann.py +++ b/adnc/model/mann.py @@ -55,7 +55,10 @@ class MANN(): if self.input_embedding: word_idx_dict = self.input_embedding['word_idx_dict'] embedding_size = self.input_embedding['embedding_size'] - tmp_dir = self.input_embedding['tmp_dir'] + if 'tmp_dir' in self.input_embedding: + tmp_dir = self.input_embedding['tmp_dir'] + else: + tmp_dir = "data_tmp" glove = WordEmbedding(embedding_size, word_idx_dict=word_idx_dict, initialization='glove', tmp_dir=tmp_dir) self._data = tf.placeholder(tf.int64, [None, self.batch_size], name='x')