auracaster/piper
3
0
Fork 0
mirror of https://github.com/pstrueb/piper.git synced 2026-05-19 20:18: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 argparse
import json import json
import time import time
import statistics
import sys import sys
import torch import torch
@@ -45,7 +46,12 @@ def main() -> None:
) )
json.dump( 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, sys.stdout,
) )

8
src/benchmark/benchmark_torchscript.py
View File

@@ -2,6 +2,7 @@
import argparse import argparse
import json import json
import time import time
import statistics
import sys import sys
import torch import torch
@@ -49,7 +50,12 @@ def main() -> None:
) )
json.dump( 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, sys.stdout,
) )