"app/git@developer.sourcefind.cn:orangecat/ollama.git" did not exist on "3f7d2336c77f09286c6ecf281fd6fa16c3ce64a5"
Commit d50d2117 authored by David E. Weekly's avatar David E. Weekly
Browse files

Fix deprecated call to tf.image_summary to tf.summary.image

parent 16da0022
......@@ -92,7 +92,7 @@ def process_image(encoded_image,
# only logged in thread 0.
def image_summary(name, image):
if not thread_id:
tf.image_summary(name, tf.expand_dims(image, 0))
tf.summary.image(name, tf.expand_dims(image, 0))
# Decode image into a float32 Tensor of shape [?, ?, 3] with values in [0, 1).
with tf.name_scope("decode", values=[encoded_image]):
......
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