{
  "seed": 42,
  "data_path": "preprocessed/dataset_al8.npz",
  "save_dir": "checkpoints_heat_v4",
  "val_split": 0.15,
  "batch_size": 64,
  "epochs": 60,
  "lr": 0.003,
  "weight_decay": 0.0001,
  "img_w": 160,
  "img_h": 120,
  "num_classes": 3,
  "out_h": 15,
  "out_w": 20,
  "stride": 8,
  "size_loss_weight": 2.0,
  "bg_weight": 0.05,
  "resume_from": "checkpoints_heat_v4/best.pth",
  "ref_sizes": {
    "0": [
      75.35844421386719,
      62.14570236206055
    ],
    "1": [
      49.79265213012695,
      38.633811950683594
    ],
    "2": [
      67.58988189697266,
      34.93575668334961
    ]
  }
}