config.json

{
  "in_dim": 10,
  "out_dim": 1,
  "hidden_dim": 64,
  "modes": 12,
  "num_layers": 4,
  "space_dim": 2,
  "include_pos": true,
  "activation": "gelu",
  "reflection": false,
  "pad_to_multiple": 16
}