diff options
Diffstat (limited to 'dnn/torch/osce/test_vocoder.py')
-rw-r--r-- | dnn/torch/osce/test_vocoder.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/dnn/torch/osce/test_vocoder.py b/dnn/torch/osce/test_vocoder.py index e71a5c37..55e5d00c 100644 --- a/dnn/torch/osce/test_vocoder.py +++ b/dnn/torch/osce/test_vocoder.py @@ -55,6 +55,7 @@ else: parser.add_argument('checkpoint', type=str, help='checkpoint file') parser.add_argument('output', type=str, help='output file') parser.add_argument('--debug', action='store_true', help='enables debug output') + parser.add_argument('--feature-version', type=int, help='feature version, default: 3', default=3) args = parser.parse_args() @@ -85,7 +86,7 @@ model.load_state_dict(checkpoint['state_dict']) # generate model input setup = checkpoint['setup'] -testdata = load_lpcnet_features(input_folder) +testdata = load_lpcnet_features(input_folder, version=args.feature_version) features = testdata['features'] periods = testdata['periods'] |