cls_inference.py

source

CLS Decoding

usage: cls_inference.py [-h] [--config CONFIG] [--log_level {CRITICAL,ERROR,WARNING,INFO,DEBUG,NOTSET}] [--batch_size BATCH_SIZE]
                        --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] [--classification_train_config CLASSIFICATION_TRAIN_CONFIG]
                        [--classification_model_file CLASSIFICATION_MODEL_FILE] [--output_all_probabilities OUTPUT_ALL_PROBABILITIES]

Named Arguments

Input data related

The model configuration related