From 1f1dee2e88158d43dbeb7d09c639d476ba43cde1 Mon Sep 17 00:00:00 2001 From: yangyaming Date: Mon, 11 Dec 2017 17:27:47 +0800 Subject: [PATCH] Adapt infer.py. --- infer.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/infer.py b/infer.py index 32d15f12..7e30549a 100644 --- a/infer.py +++ b/infer.py @@ -69,8 +69,7 @@ def infer(): augmentation_config='{}', specgram_type=args.specgram_type, num_threads=1, - keep_transcription_text=True, - num_conv_layers=args.num_conv_layers) + keep_transcription_text=True) batch_reader = data_generator.batch_reader_creator( manifest_path=args.infer_manifest, batch_size=args.num_samples, -- GitLab