Adapt infer.py.

pull/71/head
yangyaming 7 years ago
parent d7291da01f
commit 1f1dee2e88

@ -69,8 +69,7 @@ def infer():
augmentation_config='{}', augmentation_config='{}',
specgram_type=args.specgram_type, specgram_type=args.specgram_type,
num_threads=1, num_threads=1,
keep_transcription_text=True, keep_transcription_text=True)
num_conv_layers=args.num_conv_layers)
batch_reader = data_generator.batch_reader_creator( batch_reader = data_generator.batch_reader_creator(
manifest_path=args.infer_manifest, manifest_path=args.infer_manifest,
batch_size=args.num_samples, batch_size=args.num_samples,

Loading…
Cancel
Save