Hello Kamil,
The max pools are dependent on num_points which in turn depends on x.shape[1].
I wrote something like this in the init
self.maxpool = nn.MaxPool1d(1)
and used it in forward like the following:
self.maxpool = nn.MaxPool1d(num_points)
x = self.maxpool(x)
Would this work? Is it in line with your suggestion.
Best Regards,
Vijay