From b6216cd51de58d6704334bfcdfccaf3acd700f1b Mon Sep 17 00:00:00 2001
From: Triplecq <triplecq@gmail.com>
Date: Sun, 31 Mar 2024 19:53:29 -0400
Subject: [PATCH] calculate RTF

---
 egs/librispeech/ASR/zipformer/pretrained.py | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/egs/librispeech/ASR/zipformer/pretrained.py b/egs/librispeech/ASR/zipformer/pretrained.py
index de0652893..ab6216e26 100755
--- a/egs/librispeech/ASR/zipformer/pretrained.py
+++ b/egs/librispeech/ASR/zipformer/pretrained.py
@@ -111,6 +111,7 @@ Note: ./zipformer/exp/pretrained.pt is generated by ./zipformer/export.py
 import argparse
 import logging
 import math
+import time
 from typing import List
 
 import k2
@@ -314,6 +315,8 @@ def main():
     waves = [w.to(device) for w in waves]
 
     logging.info("Decoding started")
+    start_time = time.time()
+
     features = fbank(waves)
     feature_lengths = [f.size(0) for f in features]
 
@@ -366,12 +369,17 @@ def main():
             hyps.append(token_ids_to_words(hyp))
     else:
         raise ValueError(f"Unsupported method: {params.method}")
+    
+    end_time = time.time()
+    elapsed_seconds = end_time - start_time
+
+    logging.info(f"elapsed seconds:, {elapsed_seconds}")
 
     s = "\n"
     for filename, hyp in zip(params.sound_files, hyps):
         s += f"{filename}:\n{hyp}\n\n"
     logging.info(s)
-
+    
     logging.info("Decoding Done")