typo

mileyan · web-flow · commit 00e73ba9cc1f · 2018-04-10T16:39:41.000-04:00
diff --git a/main.py b/main.py
@@ -93,7 +93,7 @@ def train(imgL,imgR, disp_L):
             output3 = torch.squeeze(output3,1)
             loss = 0.5*F.smooth_l1_loss(output1[mask], disp_true[mask], size_average=True) + 0.7*F.smooth_l1_loss(output2[mask], disp_true[mask], size_average=True) + F.smooth_l1_loss(output3[mask], disp_true[mask], size_average=True) 
         elif args.model == 'basic':
-            output = model(imgL,imgR)
+            output3 = model(imgL,imgR)
             output = torch.squeeze(output3,1)
             loss = F.smooth_l1_loss(output3[mask], disp_true[mask], size_average=True)
 
