ATsurf_L/model_config

{
  "init_args": [],
  "init_kwargs": {
    "fun_dim": 3,
    "mlp_ratio": 4,
    "n_hidden": 64,
    "n_layers": 5,
    "out_dim": 3,
    "patch_size": [
      4,
      4
    ],
    "type_cond": "inj",
    "window_size": [
      8,
      8
    ]
  },
  "model_name": "WingPDETransformer",
  "schema_version": 1
}