From 631aa5d3b9c60c05c88f512717a9dad877b96ac8 Mon Sep 17 00:00:00 2001 From: Shawn Presser Date: Sun, 11 Apr 2021 23:13:57 -0700 Subject: [PATCH] Fix revnet model --- tensor2tensor/models/revnet.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tensor2tensor/models/revnet.py b/tensor2tensor/models/revnet.py index 8c923f9fe..7a78d3b89 100644 --- a/tensor2tensor/models/revnet.py +++ b/tensor2tensor/models/revnet.py @@ -114,7 +114,7 @@ def f(x, depth1, depth2, dim='2d', first_batch_norm=True, stride=1, else: net = conv(net, depth2, 3, strides=stride, padding=padding, activation=None) - net = tf.layers.batch_normalization(x, training=training) + net = tf.layers.batch_normalization(net, training=training) net = tf.nn.relu(net) net = conv(net, depth2, 3, strides=stride, padding=padding, activation=None)