cleaning code

00a2b2fc · Javier · 502de688 · 00a2b2fc
隐藏空白更改
内联并排

Showing with 0 addition and 1 deletion

wide_deep/torch_model.py wide_deep/torch_model.py +0 -1

未找到文件。
--- 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: