Commit e678fb3e authored by Cresson Remi's avatar Cresson Remi
Browse files

ENH: change deprecated TF tag constants

parent 0033b73a
......@@ -131,8 +131,8 @@ def CreateSavedModel(sess, inputs, outputs, directory):
input_dict,
output_dict)
}
builder.add_meta_graph_and_variables(sess,[tf.saved_model.tag_constants.TRAINING],signature_def_map)
builder.add_meta_graph([tf.saved_model.tag_constants.SERVING])
builder.add_meta_graph_and_variables(sess,[tf.saved_model.TRAINING],signature_def_map)
builder.add_meta_graph([tf.saved_model.SERVING])
builder.save()
def CheckpointToSavedModel(ckpt_path, inputs, outputs, savedmodel_path, clear_devices=False):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment