Skip to content

GitLab

You need to sign in or sign up before continuing.
Commit 9a173bf2 authored by George Tamer's avatar George Tamer Committed by Mengchi Zhang
Browse files

Assign self.norm_type to input norm_type

parent 164cc0f3
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
fairscale/nn/model_parallel/layers.py
View file @ 9a173bf2
...@@ -178,7 +178,7 @@ class ParallelEmbedding(torch.nn.Module): ...@@ -178,7 +178,7 @@ class ParallelEmbedding(torch.nn.Module):
self.embedding_dim = embedding_dim self.embedding_dim = embedding_dim
self.padding_idx = padding_idx self.padding_idx = padding_idx
self.max_norm = max_norm self.max_norm = max_norm
self.norm_type = scale_grad_by_freq self.norm_type = norm_type
self.scale_grad_by_freq = scale_grad_by_freq self.scale_grad_by_freq = scale_grad_by_freq
self.sparse = sparse self.sparse = sparse
self._weight = None self._weight = None
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment