Python save_tf_checkpoint Examples

Example #1

File: estimator.py Project: fuckyouNoob/analytics-zoo

    def save_tf_checkpoint(self, path):
        """
        Save tensorflow checkpoint in this estimator.

        :param path: tensorflow checkpoint path.
        """
        save_tf_checkpoint(self.sess, path)

Example #2

File: test_estimator_for_spark.py Project: zzti-bsj/analytics-zoo

    def test_checkpoint_remote(self):
        tf.reset_default_graph()

        model = SimpleModel()
        sess = tf.Session()
        sess.run(tf.global_variables_initializer())
        saver = tf.train.Saver(tf.global_variables())
        temp = tempfile.mkdtemp()
        save_tf_checkpoint(sess, os.path.join(temp, "simple.ckpt"), saver)
        ckpt = get_checkpoint_state(temp)
        assert ckpt.model_checkpoint_path == os.path.join(temp, "simple.ckpt")
        assert ckpt.all_model_checkpoint_paths[0] == os.path.join(temp, "simple.ckpt")
        load_tf_checkpoint(sess, os.path.join(temp, "simple.ckpt"), saver)
        shutil.rmtree(temp)

Example #3

 def save_tf_checkpoint(self, path):
     save_tf_checkpoint(self.sess, path)