retrina0678's picture
Add model card / clean paths
7ac90d3 verified
Raw
History Blame Contribute Delete
740 Bytes
{
"manifest": "edge_audio_runs/efficientat_target4_aug/manifest_aug_target4.csv",
"out_dir": "edge_audio_runs/efficientat_target4_aug/full_mn10_as_2s_seed42_group-source",
"efficientat_dir": "EfficientAT",
"model_name": "mn10_as",
"head_type": "mlp",
"sample_rate": 32000,
"clip_seconds": 2.0,
"epochs": 20,
"batch_size": 16,
"lr": 0.0001,
"weight_decay": 0.0001,
"n_splits": 5,
"test_size": 0.2,
"seed": 42,
"num_workers": 0,
"patience": 6,
"freeze_backbone_epochs": 2,
"amp": true,
"weighted_sampler": true,
"selection_metric": "macro_recall",
"class_weight": "balanced",
"max_samples_per_class": 0,
"group_col": "group_id",
"label_col": "label",
"path_col": "path",
"device": "auto"
}