d9421968d52736a2c1d84d26ef2c335a1be4511b,examples/pointnet++_part_seg/pointnet2_part_seg.py,PointNet2PartSegmentNet,forward,#PointNet2PartSegmentNet#Any#,226

Before Change

    def forward(self, data):
        assert hasattr(data, "pos")
        if not hasattr(data, "x"):
            data.x = None

        data_in = data.x, data.pos, data.batch

        sa1_out = self.sa1_module(data_in)
        sa2_out = self.sa2_module(sa1_out)

After Change

        fp2_out = self.fp2_module(*fp3_out, *sa1_out)
        x, _, _ = self.fp1_module(*fp2_out, *sa0_out)

        x = F.relu(self.lin1(x))
        x = F.dropout(x, training=self.training)
        x = self.lin2(x)
        x = F.dropout(x, training=self.training)
        x = self.lin3(x)
        return F.log_softmax(x, dim=-1)

Instances