amd-telephony-yamnet-gru / benchmarked_v4a /v4a_ftw_mfnone_pw05.json
ozonetg's picture
AMD telephony model: v5b_full + benchmarked v4a, code, model card
1d9e3e4 verified
Raw
History Blame Contribute Delete
564 Bytes
{
"backbone_weights": "/root/amd/tracks/yamnet/experiments/backbone/v4a_ftw_mfnone_pw05.yamnet.pth",
"head_weights": "/root/amd/tracks/yamnet/experiments/backbone/v4a_ftw_mfnone_pw05.head.pt",
"head_config": {
"proj_dim": 128,
"hidden": 96,
"num_layers": 1,
"bidirectional": true,
"pooling": "last",
"dropout": 0.0
},
"gate_dbfs": -300.0,
"decision_threshold": 0.5,
"unfreeze_blocks": 2,
"note": "load backbone via extract_embeddings.load_model(backbone_weights); build head via sweep_head.build_config_head(head_config)."
}