(self, NetClass, in_nchannel, out_nchannel, config)
| 16 | self.initialize_filter(NetClass, in_nchannel, out_nchannel, config) |
| 17 | |
| 18 | def initialize_filter(self, NetClass, in_nchannel, out_nchannel, config): |
| 19 | raise NotImplementedError("Must initialize a model and a filter") |
| 20 | |
| 21 | def forward(self, x, coords, colors=None): |
| 22 | soutput = self.model(x) |