diff --git a/tensor2tensor/layers/modalities.py b/tensor2tensor/layers/modalities.py index ddef5e67f..3dd321ca1 100644 --- a/tensor2tensor/layers/modalities.py +++ b/tensor2tensor/layers/modalities.py @@ -76,7 +76,7 @@ def _get_weights(self, hidden_dim=None): """Create or get concatenated embedding or softmax variable. Args: - hidden_dim: dim of the variable. Defaults fo self._body_input_depth + hidden_dim: dim of the variable. Defaults to self._body_input_depth Returns: a list of self._num_shards Tensors.