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