Skip to content

Commit

Permalink
merge pytorch 1.0 branch
Browse files Browse the repository at this point in the history
fix #30, #32, #27, #25, #24, #17, #14, #11, #13, #5, #4
  • Loading branch information
fxia22 committed Mar 6, 2019
2 parents b36cc4e + 610f826 commit d7a4dd5
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions pointnet/model.py
Original file line number Diff line number Diff line change
Expand Up @@ -146,6 +146,7 @@ def forward(self, x):
x = self.fc3(x)
return F.log_softmax(x, dim=1), trans, trans_feat


class PointNetDenseCls(nn.Module):
def __init__(self, k = 2, feature_transform=False):
super(PointNetDenseCls, self).__init__()
Expand Down

0 comments on commit d7a4dd5

Please sign in to comment.