fix: update scripts to support tf-2.0 (#250)

Author: chuyang-deng

test/resources/gpu_device_placement.py

@@ -19,7 +19,7 @@
     a = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[2, 3], name='a')
     b = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[3, 2], name='b')
 c = tf.matmul(a, b)
-sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
+sess = tf.compat.v1.Session(config=tf.compat.v1.ConfigProto(log_device_placement=True))
 # Runs the op.
 print(sess.run(c))
 print('-' * 87)

test/resources/keras_inception.py

@@ -26,12 +26,12 @@
 model = keras.applications.inception_v3.InceptionV3(weights='imagenet')
 
 # Exports the keras model as TensorFlow Serving Saved Model
-with tf.Session() as session:
+with tf.compat.v1.Session() as session:
 
-    init = tf.global_variables_initializer()
+    init = tf.compat.v1.global_variables_initializer()
     session.run(init)
 
-    tf.saved_model.simple_save(
+    tf.compat.v1.saved_model.simple_save(
         session,
         os.path.join(args.model_dir, 'inception-model/1'),
         inputs={'input_image': model.input},

test/resources/mnist/horovod_mnist.py

@@ -32,7 +32,7 @@
 config = tf.compat.v1.ConfigProto()
 config.gpu_options.allow_growth = True
 config.gpu_options.visible_device_list = str(hvd.local_rank())
-K.set_session(tf.Session(config=config))
+K.set_session(tf.compat.v1.Session(config=config))
 
 batch_size = 128
 num_classes = 10
@@ -119,10 +119,10 @@
     # Exports the keras model as TensorFlow Serving Saved Model
     with K.get_session() as session:
 
-        init = tf.global_variables_initializer()
+        init = tf.compat.v1.global_variables_initializer()
         session.run(init)
 
-        tf.saved_model.simple_save(
+        tf.compat.v1.saved_model.simple_save(
             session,
             os.path.join('/opt/ml/model/mnist/1'),
             inputs={'input_image': model.input},