diff --git a/examples/tiny/run.sh b/examples/tiny/run.sh index 59ecea5a2..4fa15bd53 100644 --- a/examples/tiny/run.sh +++ b/examples/tiny/run.sh @@ -6,11 +6,11 @@ source path.sh # prepare data bash ./local/run_data.sh -# test pretrain model -bash ./local/run_test_golden.sh - -# test pretain model -bash ./local/run_infer_golden.sh +## test pretrain model +#bash ./local/run_test_golden.sh +# +## test pretain model +#bash ./local/run_infer_golden.sh # train model bash ./local/run_train.sh @@ -21,5 +21,5 @@ bash ./local/run_test.sh # infer model bash ./local/run_infer.sh -# tune model -bash ./local/run_tune.sh +## tune model +#bash ./local/run_tune.sh diff --git a/infer.py b/infer.py index dc5ffe7eb..52d3a6744 100644 --- a/infer.py +++ b/infer.py @@ -13,6 +13,20 @@ # limitations under the License. """Inferer for DeepSpeech2 model.""" +import io +import logging +import argparse +import functools + +from paddle import distributed as dist + +from utils.utility import print_arguments +from training.cli import default_argument_parser + +from model_utils.config import get_cfg_defaults +from model_utils.model import DeepSpeech2Tester as Tester +from utils.error_rate import char_errors, word_errors + def main_sp(config, args): exp = Tester(config, args)