diff --git a/wide_deep/torch_model.py b/wide_deep/torch_model.py
index 3d6b85c064a8bf3f9a62ebcffb6fcd1255f63af2..bcd7c2ed9e5fffb9b63a73c0d62e8d21e9ae8a07 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: