dbaranchuk 3 lat temu
rodzic
commit
e8f84e7747
1 zmienionych plików z 1 dodań i 2 usunięć
  1. 1 2
      src/bloom/model.py

+ 1 - 2
src/bloom/model.py

@@ -584,8 +584,7 @@ class BloomForSequenceClassification(BloomPreTrainedModel):
                 )
                 )
 
 
         pooled_logits = logits[torch.arange(batch_size, device=logits.device), sequence_lengths]
         pooled_logits = logits[torch.arange(batch_size, device=logits.device), sequence_lengths]
-        pooled_logits = self.pooled_dropout(pooled_logits)
-
+        
         loss = None
         loss = None
         if labels is not None:
         if labels is not None:
             if self.config.problem_type is None:
             if self.config.problem_type is None: