Skip to content

GitLab

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
Show 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