fixed bug for yanx27#59

nidhi23ug · Mar 21, 2021 · d65f07f · d65f07f
1 parent 40ab12b
commit d65f07f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/models/pointnet_utils.py b/models/pointnet_utils.py
@@ -138,5 +138,5 @@ def feature_transform_reguliarzer(trans):
     I = torch.eye(d)[None, :, :]
     if trans.is_cuda:
         I = I.cuda()
-    loss = torch.mean(torch.norm(torch.bmm(trans, trans.transpose(2, 1) - I), dim=(1, 2)))
+    loss = torch.mean(torch.norm(torch.bmm(trans, trans.transpose(2, 1)) - I, dim=(1, 2)))
     return loss