Merge branch 'master' of...

Merge branch 'master' of https://gitlab.cl.uni-heidelberg.de/friebolin/swp-data-augmentation-for-metonymy-resolution

Code/train.py

@@ -110,6 +110,7 @@ def train(model, name,train_dataset, test_dataset, seed, batch_size, test_batch_
 								'mixepoch': True,
 								'mixlayer': mixlayer,
 								'lambda_value': lambda_value}
+						labels=batch[5]
 				
 
 			if name[0] == "r":

requirements.txt

 nltk==3.7
 numpy==1.23.5
 pandas==1.5.2
-torch==1.13.0+cu116
+torch>=1.13.0+cu116
+torchvision>=0.13.0+cu116
+torchaudio>=0.13.0
 tqdm==4.64.1
 python==3.9.*
 evaluate ==0.3.0
 matplotlib==3.5.2
 scikit_lean==1.2.1
-transformers==4.26.1
\ No newline at end of file
+transformers==4.26.1