diff --git a/tensor2tensor/layers/common_layers.py b/tensor2tensor/layers/common_layers.py index 722c29a0a..c751ad798 100644 --- a/tensor2tensor/layers/common_layers.py +++ b/tensor2tensor/layers/common_layers.py @@ -3993,7 +3993,7 @@ class WeightNorm(tf.keras.layers.Wrapper): data_init=True)(net) ``` - Arguments: + Args: layer: a layer instance. data_init: If `True` use data dependent variable initialization