diff --git a/pointnet/model.py b/pointnet/model.py index 8931fe8..e036f0b 100644 --- a/pointnet/model.py +++ b/pointnet/model.py @@ -272,7 +272,7 @@ def feature_transform_regularizer(trans, device): # https://github.com/princeton-vl/SimpleView/blob/master/models/pointnet.py class PointNet(nn.Module): - def __init__(self, dataset, task, device, feature_transform=True, attention=True): + def __init__(self, dataset, task, device, feature_transform=False, attention=True): super().__init__() self.task = task self.device = device