Upload tencent/sslmos/2337/config.yml with huggingface_hub

tencent/sslmos/2337/config.yml

@@ -0,0 +1,58 @@
+additional_config: null
+allow_cache: true
+best_model_criterion:
+  key: utt_SRCC
+  order: highest
+collater_type: NonIntrusiveCollater
+config: conf/ssl-mos-wav2vec2.yaml
+dataset_type: NonIntrusiveDataset
+dev_csv_path: data/tencent_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/tencent_train.csv
+train_max_steps: 100000
+trainer_type: NonIntrusiveEstimatorTrainer
+verbose: 1
+version: 0.1.0
+wav_only: true