Update requirements versions, change tf.compat.v1. usages to tf.

PiperOrigin-RevId: 288659456
This commit is contained in:
Augustin Zidek
2020-01-08 10:28:00 +00:00
committed by Diego de Las Casas
parent a506a4274e
commit 2160fc7f17
7 changed files with 24 additions and 23 deletions
+1 -1
View File
@@ -55,7 +55,7 @@ class Secstruct(object):
def make_layer_new(self, activations):
"""Make the layer."""
with tf.compat.v1.variable_scope(self.name, reuse=tf.AUTO_REUSE):
with tf.variable_scope(self.name, reuse=tf.AUTO_REUSE):
logging.info('Creating secstruct %s', activations)
self.logits = contrib_layers.linear(activations, self._dimension)
self.ss_q8_probs = tf.nn.softmax(self.logits)