spk_embed_extract.py

Less than 1 minute
spk_embed_extract.py

speaker embedding extraction
usage: spk_embed_extract.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 [--batch_type {unsorted,sorted,folded,length,numel}] [--batch_bins BATCH_BINS]
                            [--valid_batch_bins VALID_BATCH_BINS] [--valid_batch_type {unsorted,sorted,folded,length,numel,None}]
                            [--max_cache_size MAX_CACHE_SIZE] [--max_cache_fd MAX_CACHE_FD] [--allow_multi_rates ALLOW_MULTI_RATES]
                            [--valid_max_cache_size VALID_MAX_CACHE_SIZE] [--shape_file SHAPE_FILE] [--input_size INPUT_SIZE]
                            [--num_cohort_spk NUM_COHORT_SPK] [--num_utt_per_spk NUM_UTT_PER_SPK] [--utt_select_sec UTT_SELECT_SEC]
                            [--average_spk AVERAGE_SPK] [--adaptive_cohort_size ADAPTIVE_COHORT_SIZE] [--qmf_dur_thresh QMF_DUR_THRESH]
                            [--qmf_num_trial_per_condition QMF_NUM_TRIAL_PER_CONDITION] [--allow_variable_data_keys ALLOW_VARIABLE_DATA_KEYS]
                            [--average_embd AVERAGE_EMBD] [--train_dtype {float16,float32,float64}] [--use_amp USE_AMP]
                            [--no_forward_run NO_FORWARD_RUN] [--sort_in_batch {descending,ascending}] [--sort_batch {descending,ascending}]
                            [--drop_last_iter DROP_LAST_ITER] [--spk_train_config SPK_TRAIN_CONFIG] [--spk_model_file SPK_MODEL_FILE]
                            [--model_tag MODEL_TAG] [--dist_backend DIST_BACKEND] [--dist_init_method DIST_INIT_METHOD]
                            [--dist_world_size DIST_WORLD_SIZE] [--dist_rank DIST_RANK] [--local_rank LOCAL_RANK]
                            [--dist_master_addr DIST_MASTER_ADDR] [--dist_master_port DIST_MASTER_PORT] [--dist_launcher {slurm,mpi,None}]
                            [--multiprocessing_distributed MULTIPROCESSING_DISTRIBUTED] [--unused_parameters UNUSED_PARAMETERS]
                            [--sharded_ddp SHARDED_DDP] [--use_matplotlib USE_MATPLOTLIB] [--use_tensorboard USE_TENSORBOARD]
                            [--create_graph_in_tensorboard CREATE_GRAPH_IN_TENSORBOARD] [--use_wandb USE_WANDB] [--wandb_project WANDB_PROJECT]
                            [--wandb_id WANDB_ID] [--wandb_entity WANDB_ENTITY] [--wandb_name WANDB_NAME]
                            [--wandb_model_log_interval WANDB_MODEL_LOG_INTERVAL] [--detect_anomaly DETECT_ANOMALY] [--use_lora USE_LORA]
                            [--save_lora_only SAVE_LORA_ONLY] [--lora_conf LORA_CONF] [--cudnn_enabled CUDNN_ENABLED]
                            [--cudnn_benchmark CUDNN_BENCHMARK] [--cudnn_deterministic CUDNN_DETERMINISTIC] [--valid_batch_size VALID_BATCH_SIZE]
                            [--target_duration TARGET_DURATION] [--num_eval NUM_EVAL] [--fold_length FOLD_LENGTH]
                            [--use_preprocessor USE_PREPROCESSOR]