Method __init__

src/shape_assembly/models/decoder/MLPDecoder.py:13–21

(self, feat_dim, num_points)

class MLPDecoder(nn.Module):
    def __init__(self, feat_dim, num_points):
        super().__init__()
        self.np = num_points
        self.fc_layers = nn.Sequential(
            nn.Linear(feat_dim * 2, num_points * 2),
            nn.BatchNorm1d(num_points * 2),
            nn.LeakyReLU(0.2),
            nn.Linear(num_points * 2, num_points * 3),
        )

    def forward(self, x):
        # x.shape: (bs,1024)