csukuangfj / transducer-loss-benchmarking

Other
65 stars 10 forks source link

Add benchmark code for optimized_transducer and torchaudio. #4

Closed csukuangfj closed 2 years ago

csukuangfj commented 2 years ago

Usage:

python3 ./benchmark_ot.py
python3 ./benchmark_torchaudio.py

It will generate two folders:

To view the results, run

pip install torch-tb-profiler

and then run

tensorboard --logdir log/torchaudio-30 --port 6006

# open another terminal
tensorboard --logdir log/optimized_transducer-30 --port 6007

Open your browser and go to

You will see the following results:

Screen Shot 2022-02-27 at 21 49 34 Screen Shot 2022-02-27 at 21 50 19