File size: 1,669 Bytes
c18bc7a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 |
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-4337/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-4337/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-4337
padding_mode: repetitive
patience: 20
pin_memory: true
rank: 0
resume: ''
sampling_rate: 16000
scheduler_type: null
seed: 4337
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
|