unilight's picture
Upload bvcc+nisqa+pstn+singmos+somos+tencent+tmhint-qi/sslmos+mdf/3337/config.yml with huggingface_hub
32d5ef5 verified
additional_config: conf/ssl-mos-wav2vec2-mdf.yaml
allow_cache: false
best_model_criterion:
key: utt_SRCC
order: highest
collater_type: NonIntrusiveCollater
config: exp/ssl-mos-wav2vec2-mdf-3337/original_config.yml
dataset_type: NonIntrusiveDataset
dev_csv_path: data/dev.csv
dev_samples_per_eval_loop: 1000
distributed: false
eval_and_save_interval_steps: 1000
freeze-mods: null
grad_norm: 1.0
inference_mode: mean_net
init-mods:
- ssl_model
- mean_net_dnn
init_checkpoint: exp/ssl-mos-wav2vec2-mdf-3337/original_checkpoint-11100steps.pkl
keep_nbest_models: 5
listener_score_criterion_type: null
listener_score_criterions: null
log_interval_steps: 100
mean_score_criterion_params:
masked_loss: false
tau: 0.5
mean_score_criterion_type: ScalarLoss
mean_score_criterion_weight: 1.0
mean_score_criterions:
- criterion_params:
masked_loss: false
order: 1
tau: 0.5
criterion_type: ScalarLoss
criterion_weight: 1.0
model_input: waveform
model_params:
mean_net_dnn_dim: 64
mean_net_output_type: scalar
mean_net_range_clipping: true
s3prl_name: wav2vec2
ssl_model_layer_idx: -1
ssl_model_output_dim: 768
ssl_module: s3prl
use_listener_modeling: false
use_mean_listener: false
model_type: SSLMOS
num_domains: 7
num_workers: 0
optimizer_params:
lr: 0.001
momentum: 0.9
optimizer_type: SGD
outdir: exp/ssl-mos-wav2vec2-mdf-3337
padding_mode: repetitive
patience: 20
pin_memory: true
rank: 0
resume: ''
sampling_rate: 16000
scheduler_type: null
seed: 3337
test_batch_size: 1
train_batch_size: 16
train_csv_path: data/train.csv
train_max_steps: 100000
trainer_type: NonIntrusiveEstimatorTrainer
verbose: 1
version: 0.1.0
wav_only: true