2_Make the first model's architecture (FNO2d) more parametric

[ktatar]

Let's make the layer count a input parameter to the model.

[victorzappi]

Parametrize:
_number of stacks [spectral and regular convolution]
_number of dense layers at beginning and end of network
_number of dense layers' units
_activation function selection