speechlm_inference.py

source

SpeechLM inference

usage: speechlm_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]
                             [--batch_size BATCH_SIZE]
                             --data_path_and_name_and_type
                             DATA_PATH_AND_NAME_AND_TYPE [--key_file KEY_FILE]
                             [--train_config TRAIN_CONFIG]
                             [--model_file MODEL_FILE] [--model_tag MODEL_TAG]
                             [--maxlenratio MAXLENRATIO]
                             [--minlenratio MINLENRATIO]
                             [--search_algo {sampling,teacher_force,beam_search,greedy_search}]
                             [--nbest NBEST]
                             [--sampling_temperature SAMPLING_TEMPERATURE]
                             [--top_k TOP_K] [--inference_nj INFERENCE_NJ]

Named Arguments

Input data related

The model configuration related

Infernece related