box / config.json
eidoom's picture
add cfg
5780d2e
raw
history blame
1.27 kB
{
"device": "cpu",
"dtype": "float32",
"dimensions": {
"input": {
"variables": 1
},
"output": {
"epsilon_orders": 5,
"functions": 3
},
"connection": {
"epsilon_orders": 2
}
},
"phase_space": {
"scale": 10.0,
"const": 10.0,
"cut": {
"physical": 0.1,
"spurious": 0.1
}
},
"hyperparameters": {
"optimiser": {
"algorithm": "Adam",
"initial_learning_rate": 1e-2
},
"learning_rate_scheduler": {
"factor": 0.1,
"patience": 5,
"threshold": 1e-2,
"cooldown": 5
},
"iterations": {
"batch_size": 64,
"partition_size": 1024,
"number_partitions": 1
},
"architecture": {
"hidden_layers": [32, 32, 32],
"activation": "GELU"
},
"loss_function": {
"boundary_bias": 1.0,
"boundary_derivatives": false
}
},
"termination": {
"epochs": 0,
"seconds": 0,
"learning_rate": {
"target": 1.1e-8,
"patience": 5
}
}
}