| { | |
| "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" | |
| } |