More benchmarks

This commit is contained in:
Michael Hansen
2023-04-10 14:37:34 -05:00
parent 082e64a53c
commit c945c141f1
4 changed files with 120 additions and 12 deletions

View File

@@ -6,13 +6,13 @@ import sys
import torch
_SPEAKER_ID = 0
def main() -> None:
parser = argparse.ArgumentParser()
parser.add_argument("-m", "--model", required=True, help="Path to Onnx model file")
parser.add_argument("-c", "--config", help="Path to model config file")
parser.add_argument(
"-m", "--model", required=True, help="Path to generator file (.pt)"
)
parser.add_argument("-c", "--config", help="Path to model config file (.json)")
args = parser.parse_args()
if not args.config: