diff options
Diffstat (limited to 'tensorflow/contrib/eager/python/examples/sagan/sagan.py')
| -rw-r--r-- | tensorflow/contrib/eager/python/examples/sagan/sagan.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/contrib/eager/python/examples/sagan/sagan.py b/tensorflow/contrib/eager/python/examples/sagan/sagan.py index 561be36c91..8130414985 100644 --- a/tensorflow/contrib/eager/python/examples/sagan/sagan.py +++ b/tensorflow/contrib/eager/python/examples/sagan/sagan.py @@ -62,7 +62,7 @@ class SelfAttentionModule(tf.keras.Model): kernel_size=1, strides=(1, 1), data_format=data_format) - self.scale = tfe.Variable(0., trainable=True) + self.scale = tf.Variable(0., trainable=True) def call(self, x): f = self.f(x) |