Python generate_checkpoint_state_proto Exemples

Exemple #1

Fichier : checkpoint_management_test.py Projet : plaidml/tensorflow

 def testAbsPath(self):
     save_dir = self._get_test_dir("abs_paths")
     abs_path = os.path.join(save_dir, "model-0")
     ckpt = checkpoint_management.generate_checkpoint_state_proto(
         save_dir, abs_path)
     self.assertEqual(ckpt.model_checkpoint_path, abs_path)
     self.assertTrue(os.path.isabs(ckpt.model_checkpoint_path))
     self.assertEqual(len(ckpt.all_model_checkpoint_paths), 1)
     self.assertEqual(ckpt.all_model_checkpoint_paths[-1], abs_path)

Exemple #2

Fichier : checkpoint_management_test.py Projet : clsung/tensorflow

 def testAbsPath(self):
   save_dir = self._get_test_dir("abs_paths")
   abs_path = os.path.join(save_dir, "model-0")
   ckpt = checkpoint_management.generate_checkpoint_state_proto(
       save_dir, abs_path)
   self.assertEqual(ckpt.model_checkpoint_path, abs_path)
   self.assertTrue(os.path.isabs(ckpt.model_checkpoint_path))
   self.assertEqual(len(ckpt.all_model_checkpoint_paths), 1)
   self.assertEqual(ckpt.all_model_checkpoint_paths[-1], abs_path)

Exemple #3

Fichier : checkpoint_management_test.py Projet : plaidml/tensorflow

 def testRelPath(self):
     train_dir = "train"
     model = os.path.join(train_dir, "model-0")
     # model_checkpoint_path should have no "train" directory part.
     new_rel_path = "model-0"
     ckpt = checkpoint_management.generate_checkpoint_state_proto(
         train_dir, model)
     self.assertEqual(ckpt.model_checkpoint_path, new_rel_path)
     self.assertEqual(len(ckpt.all_model_checkpoint_paths), 1)
     self.assertEqual(ckpt.all_model_checkpoint_paths[-1], new_rel_path)

Exemple #4

Fichier : checkpoint_management_test.py Projet : clsung/tensorflow

 def testRelPath(self):
   train_dir = "train"
   model = os.path.join(train_dir, "model-0")
   # model_checkpoint_path should have no "train" directory part.
   new_rel_path = "model-0"
   ckpt = checkpoint_management.generate_checkpoint_state_proto(
       train_dir, model)
   self.assertEqual(ckpt.model_checkpoint_path, new_rel_path)
   self.assertEqual(len(ckpt.all_model_checkpoint_paths), 1)
   self.assertEqual(ckpt.all_model_checkpoint_paths[-1], new_rel_path)

Exemple #5

Fichier : checkpoint_management_test.py Projet : plaidml/tensorflow

 def testAllModelCheckpointPaths(self):
     save_dir = self._get_test_dir("all_models_test")
     abs_path = os.path.join(save_dir, "model-0")
     for paths in [None, [], ["model-2"]]:
         ckpt = checkpoint_management.generate_checkpoint_state_proto(
             save_dir, abs_path, all_model_checkpoint_paths=paths)
         self.assertEqual(ckpt.model_checkpoint_path, abs_path)
         self.assertTrue(os.path.isabs(ckpt.model_checkpoint_path))
         self.assertEqual(len(ckpt.all_model_checkpoint_paths),
                          len(paths) if paths else 1)
         self.assertEqual(ckpt.all_model_checkpoint_paths[-1], abs_path)

Exemple #6

Fichier : checkpoint_management_test.py Projet : clsung/tensorflow

 def testAllModelCheckpointPaths(self):
   save_dir = self._get_test_dir("all_models_test")
   abs_path = os.path.join(save_dir, "model-0")
   for paths in [None, [], ["model-2"]]:
     ckpt = checkpoint_management.generate_checkpoint_state_proto(
         save_dir, abs_path, all_model_checkpoint_paths=paths)
     self.assertEqual(ckpt.model_checkpoint_path, abs_path)
     self.assertTrue(os.path.isabs(ckpt.model_checkpoint_path))
     self.assertEqual(
         len(ckpt.all_model_checkpoint_paths), len(paths) if paths else 1)
     self.assertEqual(ckpt.all_model_checkpoint_paths[-1], abs_path)