From 864914f9a97cd6df158f155caa2ec65f6269439d Mon Sep 17 00:00:00 2001 From: marcoyang Date: Mon, 8 Apr 2024 18:56:19 +0800 Subject: [PATCH] update comments --- egs/audioset/AT/zipformer/train.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/egs/audioset/AT/zipformer/train.py b/egs/audioset/AT/zipformer/train.py index 917c9d9a3..0e234c59f 100644 --- a/egs/audioset/AT/zipformer/train.py +++ b/egs/audioset/AT/zipformer/train.py @@ -648,7 +648,9 @@ def compute_loss( feature = feature.to(device) 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 = labels.to(device)