fix: custom loss using inplace operations

wonjae124 · Nov 8, 2021 · 423f7c4 · 423f7c4
1 parent c88bbf5
commit 423f7c4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pytorch_tabnet/abstract_model.py b/pytorch_tabnet/abstract_model.py
@@ -471,7 +471,7 @@ def _train_batch(self, X, y):
 
         loss = self.compute_loss(output, y)
         # Add the overall sparsity loss
-        loss -= self.lambda_sparse * M_loss
+        loss = loss - self.lambda_sparse * M_loss
 
         # Perform backward pass and optimization
         loss.backward()