refactor(model): isolate attr

webpolis · Aug 26, 2023 · f1610c6 · f1610c6
1 parent 1668148
commit f1610c6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/model/model.py b/src/model/model.py
@@ -561,7 +561,7 @@ def forward(self, idx):
 
             self.register_buffer('emb_input', idx.detach().clone(), persistent=False)
             self.register_buffer('emb_output', output, persistent=False)
-            self.register_buffer('emb_hat', emb_hat, persistent=False)
+            self.register_buffer('emb_hat', emb_hat.detach().clone(), persistent=False)
             self.register_buffer('emb_orig', emb, persistent=False)
             self.register_buffer('emb_hidden', hidden, persistent=False)
             self.register_buffer('emb_mean', mean, persistent=False)