Fix deprecated options

purboo · Aug 31, 2018 · 7fbbfae · 7fbbfae
1 parent 0ad1e9c
commit 7fbbfae
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/one_shot_learning_network.py b/one_shot_learning_network.py
@@ -109,7 +109,7 @@ def __call__(self, support_set, input_image, name, training=False):
             eps = 1e-10
             similarities = []
             for support_image in tf.unstack(support_set, axis=0):
-                sum_support = tf.reduce_sum(tf.square(support_image), 1, keep_dims=True)
+                sum_support = tf.reduce_sum(tf.square(support_image), 1, keepdims=True)
                 support_magnitude = tf.rsqrt(tf.clip_by_value(sum_support, eps, float("inf")))
                 dot_product = tf.matmul(tf.expand_dims(input_image, 1), tf.expand_dims(support_image, 2))
                 dot_product = tf.squeeze(dot_product, [1, ])

diff --git a/train_one_shot_learning_matching_network.py b/train_one_shot_learning_matching_network.py
@@ -1,3 +1,5 @@
+#!/usr/bin/env python3
+
 from one_shot_learning_network import *
 from experiment_builder import ExperimentBuilder
 import tensorflow.contrib.slim as slim