config.json

{
  "model": {
    "attention": "linearno",
    "linearno_variant": "shared_qk",
    "linearno_project_out": false,
    "space_dim": 2,
    "fun_dim": 0,
    "out_dim": 3,
    "n_layers": 8,
    "n_hidden": 128,
    "n_heads": 8,
    "dim_head": 16,
    "slice_num": 64,
    "mlp_ratio": 1,
    "dropout": 0.0,
    "unified_pos": false,
    "ref": 8
  },
  "normalizer": {
    "mean": 187.73873901367188,
    "std": 127.08287811279297
  },
  "scale_S": 187.73873901367188,
  "metrics": {
    "test_rel_l2": 0.005968,
    "best_test_rel_l2": 0.005968,
    "test_residual": 16361.362044798194,
    "test_residual_scaled": 0.46420697510242465,
    "scale_S": 187.73873901367188,
    "train_rel_l2": 0.004208,
    "n_params": 582275,
    "epochs": 500,
    "approach": "A",
    "lambda": 0.01
  },
  "task": "geo-fno-elasticity-stress",
  "license": "mit"
}