Homogenise param names across models

[radka-j]

For example, we have different names for hidden params:

• hidden_channels (FNO)
• hidden_dim (ViT)
• innit_features (UNet)

We should ensure the naming is consistent to avoid confusion.