Update modeling_wav2vec2_spkreg.py

modeling_wav2vec2_spkreg.py

@@ -660,7 +660,7 @@ class Wav2Vec2SpkRegForSequenceClassification(Wav2Vec2SpkRegPreTrainedModel):
             self.classifier = nn.Linear(config.classifier_proj_size, config.num_labels)
         elif self.config.loss_fct == 'additive_margin':
             self.classifier = AngularLinear(config.classifier_proj_size, config.num_labels)
-        elif self.config.loss_fct == 'additive_margin':
+        elif self.config.loss_fct == 'additive_angular_margin':
             self.classifier = AngularLinear(config.classifier_proj_size, config.num_labels)
         else:
             raise ValueError(f"Unsupported loss function: {self.config.loss_fct}")