File size: 1,513 Bytes
7756f26 005277e 7756f26 4fd2f2d 7756f26 4fd2f2d 7756f26 005277e 4fd2f2d 005277e 4fd2f2d 005277e 7756f26 4fd2f2d 7756f26 4fd2f2d 7756f26 4fd2f2d 7756f26 4fd2f2d 7756f26 4fd2f2d 7756f26 4fd2f2d 7756f26 4fd2f2d 7756f26 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 | {
"bundle": {
"bundle_subdir": "bundle"
},
"data": {
"columns": null,
"dtype": "float32",
"inputs": [],
"memmap": true,
"path": "",
"targets": []
},
"hub": {
"repo_id": null,
"revision": null
},
"io": {
"x_dim": 3,
"x_names": [
"age",
"eep",
"feh"
],
"y_dim": 3,
"y_names": [
"G_mag",
"BP_mag",
"RP_mag"
]
},
"model": {
"name": "mlp",
"params": {
"activation": "gelu",
"dtype": "float32",
"hidden_sizes": [
64,
64,
64
]
}
},
"optim": {
"b1": 0.9,
"b2": 0.999,
"decay_steps": 0,
"eps": 1e-08,
"lr": 0.003,
"name": "soap",
"precondition_1d": false,
"precondition_frequency": 10,
"schedule": "cosine",
"warmup_steps": 10000,
"weight_decay": 1e-05
},
"schema_version": 1,
"seed": 0,
"task": {
"name": "regression",
"params": {
"loss": "mse",
"metric_axes": {
"global": "all",
"per_dim": []
},
"metrics": [
"mse",
"mae"
]
}
},
"training": {
"batch_size": 2048,
"checkpoint_every_steps": 150000,
"eval_every_steps": 5000,
"log_every_steps": 1000,
"max_checkpoints": 3,
"num_steps": 100000,
"shuffle": true,
"shuffle_seed": 0,
"steps_per_epoch": null,
"val_fraction": 0.1,
"workdir": "/Users/tr/repos/astro-emulators-toolkit/large_local_examples/isochrones/run2"
}
} |