maxmustermann/DeepLearningExamples
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

fix for github issue 965

Browse source
This commit is contained in:
gkarch 2021-07-14 17:16:10 +02:00 committed by Krzysztof Kudrynski
parent 33110132cc
commit 5be8a3cae2
1 changed files with 3 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
PyTorch/SpeechRecognition/Jasper/inference.py
View file

@ -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',
@ -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)