new print

Code/train.py

@@ -105,6 +105,7 @@ def train(model, name, seed,gradient_accumulation_steps,mixup, threshold, lambda
 						
 					print("logits: ", logits) 
 					print("logits: ", logits.size())
+					print("Newlabels: ", new_labels_batch)
 					print("labels size: ", new_labels_batch.size())
 
 					input = torch.autograd.Variable(logits.view(-1, 2).to("cuda"))