From 00a2b2fc337c5e864c4cfc174e15733226475f7a Mon Sep 17 00:00:00 2001
From: Javier <jrzaurin@gmail.com>
Date: Mon, 18 Mar 2019 13:10:55 +0000
Subject: [PATCH] cleaning code

---
 wide_deep/torch_model.py | 1 -
 1 file changed, 1 deletion(-)

diff --git a/wide_deep/torch_model.py b/wide_deep/torch_model.py
index 3d6b85c..bcd7c2e 100644
--- a/wide_deep/torch_model.py
+++ b/wide_deep/torch_model.py
@@ -269,7 +269,6 @@ class WideDeep(nn.Module):
 
                 self.optimizer.zero_grad()
                 y_pred =  net(X_w, X_d)
-                loss = None
                 if(self.criterion == F.cross_entropy):
                     loss = self.criterion(y_pred, y) #[batch_size, 1]
                 else:
-- 
GitLab