b18167d070cc0563998de54d54c8bd6757dfe7b3,art/classifiers/tensorflow.py,TFClassifier,save,#TFClassifier#Any#Any#,456

Before Change

        if not os.path.exists(folder):
            os.makedirs(folder)

        saver = tf.train.Saver()
        saver.save(self._sess, full_path)
        logger.info("Model saved in path: %s.", full_path)

    def __repr__(self):
        repr_ = "%s(clip_values=%r, input_ph=%r, logits=%r, output_ph=%r, train=%r, loss=%r, learnign=%r, " \

After Change

        if os.path.exists(full_path):
            shutil.rmtree(full_path)

        builder = saved_model.builder.SavedModelBuilder(full_path)
        signature = predict_signature_def(inputs={"SavedInputPhD": self._input_ph},
                                          outputs={"SavedOutputLogit": self._logits})
        builder.add_meta_graph_and_variables(sess=self._sess, tags=[tag_constants.SERVING],
                                             signature_def_map={"predict": signature})
        builder.save()

        logger.info("Model saved in path: %s.", full_path)

    def __repr__(self):

Instances