CEL

Code/train.py

@@ -142,7 +142,7 @@ def train(model, name, seed,gradient_accumulation_steps,mixup, threshold, lambda
 #log base e
 #Fkt vom Meeting
 def cross_entropy(logits, target, l):
-	result = torch.tensor([])
+	result = torch.tensor([], device='cuda')
 	for i in range (logits.shape[0]):
 		lg = logits[i:i+1,:]
 		t = target[i]