diff options
Diffstat (limited to 'tensorflow/contrib/layers/python/layers/regularizers_test.py')
| -rw-r--r-- | tensorflow/contrib/layers/python/layers/regularizers_test.py | 14 |
1 files changed, 7 insertions, 7 deletions
diff --git a/tensorflow/contrib/layers/python/layers/regularizers_test.py b/tensorflow/contrib/layers/python/layers/regularizers_test.py index 07191eeda7..51faba30c7 100644 --- a/tensorflow/contrib/layers/python/layers/regularizers_test.py +++ b/tensorflow/contrib/layers/python/layers/regularizers_test.py @@ -71,7 +71,7 @@ class RegularizerTest(test.TestCase): with self.assertRaises(ValueError): regularizers.l1_l2_regularizer(0.5, 0) - with self.test_session(): + with self.cached_session(): shape = [5, 5, 5] num_elem = 5 * 5 * 5 tensor = constant_op.constant(1.0, shape=shape) @@ -84,7 +84,7 @@ class RegularizerTest(test.TestCase): num_elem = 5 * 5 * 5 tensor = constant_op.constant(1.0, shape=shape) loss = regularizers.l1_l2_regularizer(0.0, 1.0)(tensor) - with self.test_session(): + with self.cached_session(): self.assertEquals(loss.op.name, 'l1_l2_regularizer') self.assertAlmostEqual(loss.eval(), num_elem / 2, 5) @@ -93,7 +93,7 @@ class RegularizerTest(test.TestCase): num_elem = 5 * 5 * 5 tensor = constant_op.constant(1.0, shape=shape) loss = regularizers.l1_l2_regularizer(1.0, 0.0)(tensor) - with self.test_session(): + with self.cached_session(): self.assertEquals(loss.op.name, 'l1_l2_regularizer') self.assertAlmostEqual(loss.eval(), num_elem, 5) @@ -104,7 +104,7 @@ class RegularizerTest(test.TestCase): self.assertEquals(loss, None) def testL1L2RegularizerWithScope(self): - with self.test_session(): + with self.cached_session(): shape = [5, 5, 5] num_elem = 5 * 5 * 5 tensor = constant_op.constant(1.0, shape=shape) @@ -142,7 +142,7 @@ class RegularizerTest(test.TestCase): array_weights_list = [[1.5], [2, 3, 4.2], [10, 42, 666.6]] tensor_weights_list = [constant_op.constant(x) for x in array_weights_list] expected = sum([2 * x for l in array_weights_list for x in l]) - with self.test_session(): + with self.cached_session(): result = regularizers.apply_regularization(dummy_regularizer, tensor_weights_list) self.assertAllClose(expected, result.eval()) @@ -151,7 +151,7 @@ class RegularizerTest(test.TestCase): regularizer = regularizers.l2_regularizer(0.0) array_weights_list = [[1.5], [2, 3, 4.2], [10, 42, 666.6]] tensor_weights_list = [constant_op.constant(x) for x in array_weights_list] - with self.test_session(): + with self.cached_session(): result = regularizers.apply_regularization(regularizer, tensor_weights_list) self.assertAllClose(0.0, result.eval()) @@ -161,7 +161,7 @@ class RegularizerTest(test.TestCase): tensor_weights_list = [ constant_op.constant(x) for x in [[1.5], [2, 3, 4.2], [10, 42, 666.6]] ] - with self.test_session(): + with self.cached_session(): with self.assertRaises(ValueError): regularizers.apply_regularization(non_scalar_regularizer, tensor_weights_list) |