diff options
Diffstat (limited to 'models/sparse.py')
-rw-r--r-- | models/sparse.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/models/sparse.py b/models/sparse.py index bcb2897..07b3413 100644 --- a/models/sparse.py +++ b/models/sparse.py | |||
@@ -15,7 +15,7 @@ class PseudoSparseEmbedding(nn.Module): | |||
15 | if dropout_p > 0.0: | 15 | if dropout_p > 0.0: |
16 | self.dropout = nn.Dropout(p=dropout_p) | 16 | self.dropout = nn.Dropout(p=dropout_p) |
17 | else: | 17 | else: |
18 | self.dropout = lambda x: x | 18 | self.dropout = nn.Identity() |
19 | 19 | ||
20 | self.register_buffer('mapping', torch.zeros(0, device=device, dtype=torch.long)) | 20 | self.register_buffer('mapping', torch.zeros(0, device=device, dtype=torch.long)) |
21 | 21 | ||