Update hyperparams.yalm

hyperparams.yalm

@@ -52,7 +52,7 @@ encoder_dims: 768
 n_classes: 5
 
 # Wav2vec2 encoder
-wav2vec2: !new:speechbrain.lobes.models.huggingface_wav2vec.HuggingFaceWav2Vec2
+embedding_model: !new:speechbrain.lobes.models.huggingface_wav2vec.HuggingFaceWav2Vec2
     source: !ref <wav2vec2_hub>
     output_norm: True
     freeze: !ref <freeze_wav2vec2>
@@ -65,7 +65,7 @@ avg_pool: !new:speechbrain.nnet.pooling.StatisticsPooling
     return_std: False
 
 
-label_lin: !new:speechbrain.nnet.linear.Linear
+classifier: !new:speechbrain.nnet.linear.Linear
     input_size: !ref <encoder_dims>
     n_neurons: !ref <n_classes>
     bias: False
@@ -121,8 +121,8 @@ lr_annealing_wav2vec2: !new:speechbrain.nnet.schedulers.NewBobScheduler
 checkpointer: !new:speechbrain.utils.checkpoints.Checkpointer
     checkpoints_dir: !ref <save_folder>
     recoverables:
-        model: !ref <model>
-        wav2vec2: !ref <wav2vec2>
+        model: !ref <classifier>
+        wav2vec2: !ref <embedding_model>
         lr_annealing_output: !ref <lr_annealing>
         lr_annealing_wav2vec2: !ref <lr_annealing_wav2vec2>
         counter: !ref <epoch_counter>