tree 7f8592949dc7dff5b1403ca040d5d9abb9195866
parent 796f522c803386a9ac14e5e819f4b2d71301adcd
author Scott Zhu <scottzhu@google.com> 1594402886 -0700
committer TensorFlower Gardener <gardener@tensorflow.org> 1594404550 -0700

Fix the test order dependency issue for gradient_checkpointing.

The root cause is probably related to keras global graph and some potential memory leak. Adding a teardown method to force clean any keras model created in the test, with a force GC.

Also cleanup some assertion methods which make the test code more readable.

PiperOrigin-RevId: 320630906
Change-Id: Ib72a39c184613e431d6eab03b18f3b217cb8e506