(self)
| 1443 | gfile.Exists(checkpoint_management.meta_graph_filename(s3))) |
| 1444 | |
| 1445 | def testNoMaxToKeep(self): |
| 1446 | save_dir = self._get_test_dir("no_max_to_keep") |
| 1447 | save_dir2 = self._get_test_dir("max_to_keep_0") |
| 1448 | |
| 1449 | with self.cached_session() as sess: |
| 1450 | v = variables.VariableV1(10.0, name="v") |
| 1451 | self.evaluate(variables.global_variables_initializer()) |
| 1452 | |
| 1453 | # Test max_to_keep being None. |
| 1454 | save = saver_module.Saver({"v": v}, max_to_keep=None) |
| 1455 | self.assertEqual([], save.last_checkpoints) |
| 1456 | s1 = save.save(sess, os.path.join(save_dir, "s1")) |
| 1457 | self.assertEqual([], save.last_checkpoints) |
| 1458 | self.assertTrue(checkpoint_management.checkpoint_exists(s1)) |
| 1459 | s2 = save.save(sess, os.path.join(save_dir, "s2")) |
| 1460 | self.assertEqual([], save.last_checkpoints) |
| 1461 | self.assertTrue(checkpoint_management.checkpoint_exists(s2)) |
| 1462 | |
| 1463 | # Test max_to_keep being 0. |
| 1464 | save2 = saver_module.Saver({"v": v}, max_to_keep=0) |
| 1465 | self.assertEqual([], save2.last_checkpoints) |
| 1466 | s1 = save2.save(sess, os.path.join(save_dir2, "s1")) |
| 1467 | self.assertEqual([], save2.last_checkpoints) |
| 1468 | self.assertTrue(checkpoint_management.checkpoint_exists(s1)) |
| 1469 | s2 = save2.save(sess, os.path.join(save_dir2, "s2")) |
| 1470 | self.assertEqual([], save2.last_checkpoints) |
| 1471 | self.assertTrue(checkpoint_management.checkpoint_exists(s2)) |
| 1472 | |
| 1473 | def testNoMetaGraph(self): |
| 1474 | save_dir = self._get_test_dir("no_meta_graph") |
nothing calls this directly
no test coverage detected