set TOLERANCE_ABSOLUTE
This commit is contained in:
Родитель
d5a18ba480
Коммит
025f24ce14
|
@ -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]
|
||||
|
|
Загрузка…
Ссылка в новой задаче