print statements

Code/train.py

@@ -146,6 +146,7 @@ def train(model, name, seed,gradient_accumulation_steps,mixup, threshold, lambda
 #pytorch Forum, try with dim=1 for targets
 def cross_entropy(logits, target):
     logprobs = torch.nn.functional.log_softmax (logits, dim = 1)
+    print("target for cross entropy: ", target)
     loss = -torch.sum(target * logprobs, dim=1)
     print("Logits Shape Index 0: ", logits.shape[0])
     return loss.div(logits.shape[0])