config.json

{
    "device": "cpu",
    "dtype": "float32",
    "dimensions": {
        "input": {
            "variables": 2
        },
        "output": {
            "functions": 18,
            "epsilon_orders": 5
        },
        "connection": {
            "epsilon_orders": 2
        }
    },
    "phase_space": {
        "scale": 2.5,
        "const": 2.5,
        "cut": {
            "physical": 0.1,
            "spurious": 0.1
        }
    },
    "hyperparameters": {
        "optimiser": {
            "algorithm": "Adam",
            "initial_learning_rate": 1e-3
        },
        "learning_rate_scheduler": {
            "factor": 0.1,
            "patience": 6,
            "threshold": 1e-2,
            "cooldown": 0
        },
        "iterations": {
            "batch_size": 256,
            "partition_size": 512,
            "number_partitions": 1
        },
        "architecture": {
            "hidden_layers": [256, 256, 256],
            "activation": "GELU"
        },
        "loss_function": {
            "boundary_bias": 1.0,
            "boundary_derivatives": false
        }
    },
    "termination": {
        "epochs": 0,
        "seconds": 0,
        "learning_rate": {
            "target": 1.1e-8,
            "patience": 10
        }
    },
    "zeros": {
        "re": [
            [0, 0],
            [2, 0],
            [4, 0],
            [5, 0],
            [5, 1],
            [7, 0],
            [8, 0],
            [9, 0],
            [9, 1],
            [11, 0],
            [11, 1],
            [12, 0],
            [12, 1],
            [13, 0],
            [13, 1],
            [13, 2],
            [13, 3],
            [14, 0],
            [14, 1],
            [14, 2],
            [15, 0],
            [15, 1],
            [16, 0],
            [16, 1],
            [16, 2]
        ],
        "im": [
            [0, 0],
            [0, 1],
            [1, 0],
            [1, 1],
            [2, 0],
            [2, 1],
            [3, 0],
            [3, 1],
            [3, 2],
            [3, 3],
            [4, 0],
            [4, 1],
            [4, 2],
            [5, 0],
            [5, 1],
            [6, 0],
            [7, 0],
            [8, 0],
            [9, 0],
            [9, 1],
            [9, 2],
            [10, 0],
            [10, 1],
            [10, 2],
            [10, 3],
            [10, 4],
            [11, 0],
            [11, 1],
            [12, 0],
            [12, 1],
            [12, 2],
            [13, 0],
            [13, 1],
            [13, 2],
            [13, 3],
            [13, 4],
            [14, 0],
            [14, 1],
            [14, 2],
            [15, 0],
            [15, 1],
            [16, 0],
            [16, 1],
            [16, 2],
            [16, 3],
            [17, 0]
        ]
    }
}