mirror of
https://github.com/google-deepmind/deepmind-research.git
synced 2026-05-10 05:17:46 +08:00
Update requirements versions, change tf.compat.v1. usages to tf.
PiperOrigin-RevId: 288659456
This commit is contained in:
committed by
Diego de Las Casas
parent
a506a4274e
commit
2160fc7f17
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user