33 lines
615 B
JSON
33 lines
615 B
JSON
{
|
|
"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
|
|
]
|
|
}
|
|
} |