|
|
|
@ -16,12 +16,17 @@ import os
|
|
|
|
|
import unittest
|
|
|
|
|
os.environ['FLAGS_eager_delete_tensor_gb'] = "0.0"
|
|
|
|
|
|
|
|
|
|
# FIXME(zjl): It seems that this unittest fails randomly
|
|
|
|
|
# when comparing all reduce last loss and reduce last loss
|
|
|
|
|
# e.g.: AssertionError: 1.0357145 != 1.0673475 within 0.01 delta
|
|
|
|
|
# Disable it temporarily.
|
|
|
|
|
'''
|
|
|
|
|
from test_parallel_executor_mnist import TestMNIST
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class EagerDeletionTestMNIST(TestMNIST):
|
|
|
|
|
pass
|
|
|
|
|
|
|
|
|
|
'''
|
|
|
|
|
|
|
|
|
|
if __name__ == '__main__':
|
|
|
|
|
unittest.main()
|
|
|
|
|