Исправлен путь к модели

model_SingleLabelClassifier.py

@@ -25,7 +25,7 @@ class SingleLabelClassifier(nn.Module):
             token_type_ids=token_type_ids,
             return_dict=True
         )
-        pooled_output = outputs.pooler_output
+        pooled_output = outputs.last_hidden_state[:, 0]
         x = self.intermediate(pooled_output)
         x = self.norm(x)
         x = self.activation(x)