additional_config: null | |
allow_cache: true | |
best_model_criterion: | |
key: sys_SRCC | |
order: highest | |
collater_type: NonIntrusiveCollater | |
config: conf/ssl-mos-wav2vec2.yaml | |
dataset_type: NonIntrusiveDataset | |
dev_csv_path: data/dev.csv | |
distributed: false | |
eval_and_save_interval_steps: 100 | |
grad_norm: 1.0 | |
inference_mode: mean_net | |
init_checkpoint: '' | |
keep_nbest_models: 5 | |
listener_score_criterions: null | |
log_interval_steps: 100 | |
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_workers: 0 | |
optimizer_params: | |
lr: 0.001 | |
momentum: 0.9 | |
optimizer_type: SGD | |
outdir: exp/ssl-mos-wav2vec2-2337 | |
padding_mode: repetitive | |
patience: 20 | |
pin_memory: true | |
rank: 0 | |
resume: '' | |
sampling_rate: 16000 | |
scheduler_type: null | |
seed: 2337 | |
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 | |