push

Code/train.py

@@ -137,7 +137,7 @@ def train(model, name, seed,gradient_accumulation_steps,mixup, threshold, lambda
 
 	return evaluation_test, evaluation_train
 
-#pytorch Forum, try with dim=1 for targets, changed from /logits.shape[0] zu .mean
+#pytorch Forum, try with dim=1 for targets, changed from /logits.shape[0] zu .mean..
 # def cross_entropy(logits, target):
 # 	print("Cross Entropy Target: ", target)
 # 	logprobs = torch.nn.functional.log_softmax (logits, dim = 1)