Bladeren bron

[Optimization] fix bug

wanchen.swc 2 jaren geleden
bovenliggende
commit
6f56bed4eb
1 gewijzigde bestanden met toevoegingen van 2 en 2 verwijderingen
  1. 2 2
      funasr/runtime/python/utils/test_rtf_gpu.py

+ 2 - 2
funasr/runtime/python/utils/test_rtf_gpu.py

@@ -29,11 +29,11 @@ wav_files = wav_file_f.readlines()
 
 
 # warm-up
 # warm-up
 total = 0.0
 total = 0.0
-num = 30
+num = 100
 wav_path = []
 wav_path = []
 for i in range(num * args.batch_size):
 for i in range(num * args.batch_size):
     wav_path_i = wav_files[i % len(wav_files)]
     wav_path_i = wav_files[i % len(wav_files)]
-    wav_path_i = wav_path_i.split("\t")[1].strip() if "\t" in wav_files[0] else wav_files[0].split(" ")[1].strip()
+    wav_path_i = wav_path_i.split("\t")[1].strip() if "\t" in wav_path_i else wav_path_i.split(" ")[1].strip()
     wav_path += [wav_path_i]
     wav_path += [wav_path_i]
 beg_time = time.time()
 beg_time = time.time()
 result = model(wav_path)
 result = model(wav_path)