Skip to content

Commit

Permalink
fix for github issue 965
Browse files Browse the repository at this point in the history
  • Loading branch information
grzegorz-k-karch authored and nv-kkudrynski committed Jul 15, 2021
1 parent 3311013 commit 5be8a3c
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion PyTorch/SpeechRecognition/Jasper/inference.py
Original file line number Diff line number Diff line change
Expand Up @@ -57,6 +57,8 @@ def get_parser():
help='Relative path to evaluation dataset manifest files')
parser.add_argument('--ckpt', default=None, type=str,
help='Path to model checkpoint')
parser.add_argument("--max_duration", default=None, type=float, help='maximum duration of sequences. if None uses attribute from model configuration file')
parser.add_argument("--pad_to_max_duration", action='store_true', help='pad to maximum duration of sequences')
parser.add_argument('--amp', '--fp16', action='store_true',
help='Use FP16 precision')
parser.add_argument('--cudnn_benchmark', action='store_true',
Expand Down Expand Up @@ -247,7 +249,7 @@ def main():
config_features=features_kw,
json_names=args.val_manifests,
batch_size=args.batch_size,
pipeline_type=("train" if measure_perf else "val"), # no drop_last
pipeline_type=("train" if measure_perf else "val"), # no drop_last
device_type=args.dali_device,
symbols=symbols)

Expand Down

0 comments on commit 5be8a3c

Please sign in to comment.