geometric_classifier/training_config.json

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