This commit is contained in:
StillKeepTry 2017-07-24 02:10:24 -07:00
Родитель d5a18ba480
Коммит 025f24ce14
2 изменённых файлов: 2 добавлений и 2 удалений

Просмотреть файл

@ -79,7 +79,7 @@ class DataSource(UserMinibatchSource):
self.label1 = StreamInformation("label1", 2, 'sparse', np.float32, (self.vocab_base,))
self.label2 = StreamInformation("label2", 3, 'sparse', np.float32, (self.vocab_base,))
self.word1 = StreamInformation("word1", 4, 'dense', np.float32, (1,))
self.word2 = StreamInformation("word1", 5, 'dense', np.float32, (1,))
self.word2 = StreamInformation("word2", 5, 'dense', np.float32, (1,))
super(DataSource, self).__init__()

Просмотреть файл

@ -22,7 +22,7 @@ sys.path.append(example_dir)
script_under_test = os.path.join(example_dir, 'train.py')
TIMEOUT_SECONDS = 300
TOLERANCE_ABSOLUTE = 1E-2
TOLERANCE_ABSOLUTE = 1E-1
def run_command(**kwargs):
command = ['mpiexec', '-n', '1', 'python', script_under_test]