(self, in_channels, out_channels)
| 95 | |
| 96 | class PointNetFPModule(nn.Module): |
| 97 | def __init__(self, in_channels, out_channels): |
| 98 | super().__init__() |
| 99 | self.mlp = SharedMLP(in_channels=in_channels, out_channels=out_channels, dim=1) |
| 100 | |
| 101 | def forward(self, inputs): |
| 102 | if len(inputs) == 3: |