bug fix

cfzd · cfzd · commit 2cc064e1c055 · 2019-01-22T03:34:56.000+08:00
diff --git a/PSACUDA/functions/PSANetFunc.py b/PSACUDA/functions/PSANetFunc.py
@@ -29,7 +29,7 @@ def backward(ctx,mask_grad):
 
 		bottom1_grad = torch.zeros(b1_grad_n,b1_grad_c,b1_grad_h,b1_grad_w).cuda()
 	
-		PSAWapper.PSA_backward(1,bottom1_grad, mask_grad.data);
+		PSAWapper.PSA_backward(1,bottom1_grad, mask_grad.data)
 		
 		return Variable(bottom1_grad)
 
@@ -58,6 +58,6 @@ def backward(ctx,mask_grad):
 
 		bottom1_grad = torch.zeros(b1_grad_n,b1_grad_c,b1_grad_h,b1_grad_w).cuda()
 	
-		PSAWapper.PSA_backward(2,bottom1_grad, mask_grad.data);
+		PSAWapper.PSA_backward(2,bottom1_grad, mask_grad.data)
 		
 		return Variable(bottom1_grad)
