From 97445e8111bf3e3fec831a28eab2e7f2c4d4c07c Mon Sep 17 00:00:00 2001 From: TITAN-PyCompat Date: Thu, 9 May 2019 23:52:01 +0800 Subject: [PATCH] fix api compatibility issue --- Season2/1 Word2Vec/word2vec_tf.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/Season2/1 Word2Vec/word2vec_tf.py b/Season2/1 Word2Vec/word2vec_tf.py index 0049ecb..f00e7a3 100644 --- a/Season2/1 Word2Vec/word2vec_tf.py +++ b/Season2/1 Word2Vec/word2vec_tf.py @@ -183,7 +183,12 @@ def generate_batch(batch_size, num_skips, skip_window): similarity = tf.matmul(valid_embeddings, normalized_embeddings, transpose_b=True) # Add variable initializer. - init = tf.global_variables_initializer() + tf_major_ver = int(tf.__version__.split(".")[0]) + tf_minor_ver = int(tf.__version__.split(".")[1]) + if(tf_major_ver==0 and tf_minor_ver<12): + init = tf.initialize_all_variables() + else: + init = tf.global_variables_initializer() # Step 5: Begin training. num_steps = 100001