auracaster/piper
3
0
Fork 0
mirror of https://github.com/pstrueb/piper.git synced 2026-05-19 03:58:01 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add stats

Browse Source
This commit is contained in:
Michael Hansen
2023-04-10 15:23:48 -05:00
parent c945c141f1
commit 92eb23197b
2 changed files with 14 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/benchmark/benchmark_generator.py
View File

@@ -2,6 +2,7 @@
import argparse
import json
import time
import statistics
import sys
import torch
@@ -45,7 +46,12 @@ def main() -> None:
)
json.dump(
{"load_sec": load_sec, "synthesize_rtf": synthesize_rtf},
{
"load_sec": load_sec,
"rtf_mean": statistics.mean(synthesize_rtf),
"rtf_stdev": statistics.stdev(synthesize_rtf),
"synthesize_rtf": synthesize_rtf,
},
sys.stdout,
)

8
src/benchmark/benchmark_torchscript.py
View File

@@ -2,6 +2,7 @@
import argparse
import json
import time
import statistics
import sys
import torch
@@ -49,7 +50,12 @@ def main() -> None:
)
json.dump(
{"load_sec": load_sec, "synthesize_rtf": synthesize_rtf},
{
"load_sec": load_sec,
"rtf_mean": statistics.mean(synthesize_rtf),
"rtf_stdev": statistics.stdev(synthesize_rtf),
"synthesize_rtf": synthesize_rtf,
},
sys.stdout,
)