From 876cd51a8843581bb223e311881296d0bfcac51e Mon Sep 17 00:00:00 2001 From: Joseph Oladokun Date: Fri, 16 Apr 2021 15:50:15 +0100 Subject: [PATCH] change tf.get_variable to tf.Variable --- addressnet/model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/addressnet/model.py b/addressnet/model.py index 85853a2..acfb2dc 100644 --- a/addressnet/model.py +++ b/addressnet/model.py @@ -18,7 +18,7 @@ def model_fn(features: Dict[str, tf.Tensor], labels: tf.Tensor, mode: str, param rnn_size = params.get("rnn_size", 128) rnn_layers = params.get("rnn_layers", 3) - embeddings = tf.get_variable("embeddings", dtype=tf.float32, initializer=tf.random_normal(shape=(len(vocab), 8))) + embeddings = tf.Variable("embeddings", dtype=tf.float32, initializer=tf.random_normal(shape=(len(vocab), 8))) encoded_strings = tf.nn.embedding_lookup(embeddings, encoded_text) logits, loss = nnet(encoded_strings, lengths, rnn_layers, rnn_size, labels, mode == tf.estimator.ModeKeys.TRAIN)