s2t_inference.py

source

S2T Decoding

usage: s2t_inference.py [-h] [--config CONFIG] [--log_level {CRITICAL,ERROR,WARNING,INFO,DEBUG,NOTSET}] --output_dir OUTPUT_DIR [--ngpu NGPU] [--seed SEED] [--dtype {float16,float32,float64}]
                        [--num_workers NUM_WORKERS] --data_path_and_name_and_type DATA_PATH_AND_NAME_AND_TYPE [--key_file KEY_FILE] [--allow_variable_data_keys ALLOW_VARIABLE_DATA_KEYS]
                        [--s2t_train_config S2T_TRAIN_CONFIG] [--s2t_model_file S2T_MODEL_FILE] [--lm_train_config LM_TRAIN_CONFIG] [--lm_file LM_FILE] [--word_lm_train_config WORD_LM_TRAIN_CONFIG]
                        [--word_lm_file WORD_LM_FILE] [--ngram_file NGRAM_FILE] [--model_tag MODEL_TAG] [--lang_sym LANG_SYM] [--task_sym TASK_SYM] [--predict_time PREDICT_TIME]
                        [--quantize_s2t_model QUANTIZE_S2T_MODEL] [--quantize_lm QUANTIZE_LM] [--quantize_modules [QUANTIZE_MODULES ...]] [--quantize_dtype {float16,qint8}] [--batch_size BATCH_SIZE] [--nbest NBEST]
                        [--beam_size BEAM_SIZE] [--penalty PENALTY] [--maxlenratio MAXLENRATIO] [--minlenratio MINLENRATIO] [--ctc_weight CTC_WEIGHT] [--lm_weight LM_WEIGHT] [--ngram_weight NGRAM_WEIGHT]
                        [--normalize_length NORMALIZE_LENGTH] [--token_type {char,bpe,word,None}] [--bpemodel BPEMODEL] [--partial_ar PARTIAL_AR] [--threshold_probability THRESHOLD_PROBABILITY]
                        [--max_seq_len MAX_SEQ_LEN] [--max_mask_parallel MAX_MASK_PARALLEL]

Named Arguments

Input data related

Model configuration related

Quantization related

Beam-search related

Text converter related

Partially AR related