mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-08-26 18:24:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

update comments

Browse Source
This commit is contained in:
marcoyang 2024-04-08 18:56:19 +08:00
parent 1ca4646562
commit 864914f9a9
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
egs/audioset/AT/zipformer/train.py
View File

@ -648,7 +648,9 @@ def compute_loss(
feature = feature.to(device) feature = feature.to(device)
supervisions = batch["supervisions"] supervisions = batch["supervisions"]
events = supervisions["audio_event"] # the label indices are in CED format events = supervisions[
"audio_event"
] # the label indices are in CED format (https://github.com/RicherMans/CED)
labels, _ = str2multihot(events, n_classes=params.num_events) labels, _ = str2multihot(events, n_classes=params.num_events)
labels = labels.to(device) labels = labels.to(device)