From 6a7ac689cf2367a11491d9f1768af51a8c4b1c58 Mon Sep 17 00:00:00 2001 From: marcoyang Date: Fri, 29 Mar 2024 17:08:16 +0800 Subject: [PATCH] minor updates --- egs/audioset/AT/zipformer/pretrained.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/egs/audioset/AT/zipformer/pretrained.py b/egs/audioset/AT/zipformer/pretrained.py index a162a8bb6..60e4d0518 100755 --- a/egs/audioset/AT/zipformer/pretrained.py +++ b/egs/audioset/AT/zipformer/pretrained.py @@ -185,14 +185,14 @@ def main(): encoder_out, encoder_out_lens = model.forward_encoder(features, feature_lengths) logits = model.forward_audio_tagging(encoder_out, encoder_out_lens) - for i, logit in enumerate(logits): + for filename, logit in zip(args.sound_files, logits): topk_prob, topk_index = logit.sigmoid().topk(5) topk_labels = [label_dict[index.item()] for index in topk_index] - print( - f"Top 5 predicted labels of the {i} th audio are {topk_labels} with probability of {topk_prob.tolist()}" + logging.info( + f"{filename}: Top 5 predicted labels are {topk_labels} with probability of {topk_prob.tolist()}" ) - logging.info("Decoding Done") + logging.info("Done") if __name__ == "__main__":