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

model_SingleLabelClassifier.py

@@ -3,7 +3,7 @@ import torch.nn as nn
 import torch
 
 class SingleLabelClassifier(nn.Module):
-    def __init__(self, base_model_name, num_labels, hidden_size=1024, freeze_bert=True):
+    def __init__(self, base_model_name, num_labels, hidden_size=2024, freeze_bert=True):
         super(SingleLabelClassifier, self).__init__()
         self.base = AutoModel.from_pretrained(base_model_name)
 
@@ -13,7 +13,7 @@ class SingleLabelClassifier(nn.Module):
                     param.requires_grad = False
 
         self.intermediate = nn.Linear(self.base.config.hidden_size, hidden_size)
-        self.norm = nn.BatchNorm1d(hidden_size)
+        self.norm = nn.LayerNorm(hidden_size)
         self.activation = nn.ReLU()
         self.dropout = nn.Dropout(0.4)
         self.classifier = nn.Linear(hidden_size, num_labels)