Internal change

PiperOrigin-RevId: 425409937

Internal change
PiperOrigin-RevId: 425409937
f7754fe5 · Marc van Zee · A. Unique TensorFlower · 441e5ae0 · f7754fe5
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

official/projects/vit/modeling/vit.py official/projects/vit/modeling/vit.py +1 -1

未找到文件。
--- 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,