diff --git a/scripts/nemo/qnn/parakeet-ctc/wrapper.py b/scripts/nemo/qnn/parakeet-ctc/wrapper.py index 2b9c54c239..02c03ccbc3 100755 --- a/scripts/nemo/qnn/parakeet-ctc/wrapper.py +++ b/scripts/nemo/qnn/parakeet-ctc/wrapper.py @@ -64,6 +64,8 @@ def main(): m = ModelWrapper(asr_model) m.eval() + feat_dim = asr_model.cfg.preprocessor.features + print("feat_dim", feat_dim) x = torch.rand(1, feat_dim, args.max_len, dtype=torch.float32)