config.json

{
  "model_type": "fela_pde_fno2d",
  "description": "battery pack thermal surrogate; 8 channel physics field in, temperature field out",
  "architecture": "fno2d",
  "framework": "pytorch",
  "arch": {
    "in_ch": 8,
    "out_ch": 1,
    "width": 32,
    "modes": 12,
    "layers": 3,
    "proj_hidden": 128,
    "grid_h": 96,
    "grid_w": 96
  },
  "params": 892545,
  "input_shape": [
    1,
    8,
    96,
    96
  ],
  "output_shape": [
    1,
    1,
    96,
    96
  ],
  "weights_safetensors": "model.safetensors",
  "input_channels": [
    "mask",
    "q_source",
    "k_field",
    "h_conv",
    "T_amb",
    "x_coord",
    "y_coord",
    "log_domain_L"
  ],
  "norm": {
    "x_mean": [
      0.4555675685405731,
      33879.58984375,
      4.985158443450928,
      58.90073013305664,
      27.50713348388672,
      0.5,
      0.5,
      -2.6601810455322266
    ],
    "x_std": [
      0.4980228543281555,
      89317.828125,
      6.717851638793945,
      51.0863151550293,
      7.186279773712158,
      0.29169902205467224,
      0.29169902205467224,
      0.3988337218761444
    ],
    "y_mean": 34.82331466674805,
    "y_std": 18.65639305114746,
    "amb_channel": 4
  },
  "interface": "Ynorm = model((X - x_mean)/x_std); Y_degC = Ynorm*y_std + y_mean",
  "license": "lowdown-labs-lovely-license-1.0",
  "architectures": [
    "FelaPdeModel"
  ],
  "auto_map": {
    "AutoConfig": "configuration_pde.FelaPdeConfig",
    "AutoModel": "modeling_pde.FelaPdeModel"
  },
  "complex_keys": [
    "sp.0.w1",
    "sp.0.w2",
    "sp.1.w1",
    "sp.1.w2",
    "sp.2.w1",
    "sp.2.w2"
  ]
}