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]