Merge pull request #14635 from reyoung/feature/loose_atol

Loose atol in test_pe_test_wwhile_train
local_add_cudnn_lstm
Yu Yang 6 years ago committed by GitHub
commit 4ffc376497
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -88,7 +88,7 @@ class ParallelExecutorTestingDuringTraining(unittest.TestCase):
self.assertTrue(
np.allclose(
train_loss, test_loss, atol=1e-8),
train_loss, test_loss, atol=1e-2),
"Train loss: " + str(train_loss) + "\n Test loss:" +
str(test_loss))

Loading…
Cancel
Save