From f7754fe5b8e8c10092d4155715ab2395580d4901 Mon Sep 17 00:00:00 2001 From: Marc van Zee Date: Mon, 31 Jan 2022 11:00:10 -0800 Subject: [PATCH] Internal change PiperOrigin-RevId: 425409937 --- official/projects/vit/modeling/vit.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/official/projects/vit/modeling/vit.py b/official/projects/vit/modeling/vit.py index d4512b652..5abfefaf2 100644 --- a/official/projects/vit/modeling/vit.py +++ b/official/projects/vit/modeling/vit.py @@ -145,7 +145,7 @@ class Encoder(tf.keras.layers.Layer): self._encoder_layers = [] # Set layer norm epsilons to 1e-6 to be consistent with JAX implementation. - # https://flax.readthedocs.io/en/latest/_autosummary/flax.nn.LayerNorm.html + # https://flax.readthedocs.io/en/latest/_autosummary/flax.deprecated.nn.LayerNorm.html for i in range(self._num_layers): encoder_layer = nn_blocks.TransformerEncoderBlock( inner_activation=activations.gelu, -- GitLab