Merge pull request JiaRenChang#14 from mileyan/patch-1

JiaRenChang · web-flow · commit 84ae7234e1cf · 2018-04-11T23:28:43.000+08:00
typo
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)
 
