@@ -58,7 +58,7 @@ class WordClassificationModel(torch.nn.Module): #AutoModel verwenden aus der Bib
self.classifier=nn.Linear(768,2)#first element: Hidden size, defaults to 768, should we change it to 512 like in email? Alo 768 in the other two classes as per default
#self.embedding_model.init_weights() #do we need to reimplement this?