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