File size: 704 Bytes
e1bf8d9 084f71b 36430b4 084f71b 36430b4 e1bf8d9 36430b4 084f71b 36430b4 e1bf8d9 |
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 |
{
"n_samples": 500000,
"epochs": 80,
"batch_size": 4096,
"lr": 0.003,
"seed": 42,
"optimizer": "AdamW",
"weight_decay": 0.0001,
"scheduler": "cosine_with_warmup",
"warmup_epochs": 5,
"amp_dtype": "torch.bfloat16",
"loss_weights": {
"cls": 1.0,
"fill": 0.3,
"peak": 0.3,
"ovf": 0.05,
"div": 0.02,
"vol": 0.1,
"cm": 0.1,
"curved": 0.2,
"ctype": 0.2,
"arb_cls": 1.0,
"arb_traj": 0.3,
"arb_conf": 0.15,
"flow": 0.8
},
"best_val_accuracy": 0.901,
"learned_capacities": [
0.0032398805487900972,
0.007025932893157005,
0.05782651901245117,
17.649370193481445
],
"total_training_time_seconds": 857.9640436172485
} |