def test_export_saved_model(self):
checkpoint_path = self.save_checkpoint()
destination_dir = os.path.join(checkpoint_path, 'export')
export_saved_model.export(checkpoint_path, destination_dir,
use_tf_sampling=True)
self.assertTrue(tf.gfile.Exists(
os.path.join(destination_dir, 'saved_model.pb')))
tf.gfile.DeleteRecursively(checkpoint_path)
def test_export_saved_model(self):
checkpoint_path = self.save_checkpoint()
destination_dir = os.path.join(checkpoint_path, 'export')
export_saved_model.export(checkpoint_path, destination_dir,
use_tf_sampling=True)
self.assertTrue(tf.gfile.Exists(
os.path.join(destination_dir, 'saved_model.pb')))
tf.gfile.DeleteRecursively(checkpoint_path)
