From a0b63ee7f4a8c793c0d200c86ef07677aa4cbf2e Mon Sep 17 00:00:00 2001
From: Volpeon <git@volpeon.ink>
Date: Thu, 13 Apr 2023 07:14:24 +0200
Subject: Update

---
 models/sparse.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'models/sparse.py')

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):
         if dropout_p > 0.0:
             self.dropout = nn.Dropout(p=dropout_p)
         else:
-            self.dropout = lambda x: x
+            self.dropout = nn.Identity()
 
         self.register_buffer('mapping', torch.zeros(0, device=device, dtype=torch.long))
 
-- 
cgit v1.2.3-54-g00ecf