Skip to content

Commit 2ef7331

Browse files
authored
Update sparse.py
1 parent b39a2f2 commit 2ef7331

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

torch/nn/modules/sparse.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@ class Embedding(Module):
1818
max_norm (float, optional): If given, will renormalize the embeddings to always have a norm lesser than this
1919
norm_type (float, optional): The p of the p-norm to compute for the max_norm option
2020
scale_grad_by_freq (boolean, optional): if given, this will scale gradients by the frequency of
21-
the words in the dictionary.
21+
the words in the mini-batch.
2222
2323
Attributes:
2424
weight (Tensor): the learnable weights of the module of shape (num_embeddings, embedding_dim)

0 commit comments

Comments
 (0)