shape

Code/train.py

@@ -104,9 +104,11 @@ def train(model, name, seed,gradient_accumulation_steps,mixup, threshold, lambda
 					logits=model.classifier(span_output.detach()) #target_value?
 						
 					print("logits: ", logits) 
-					print("logits: ", logits.size())
+					print("logits size: ", logits.size())
+					print("logits shape: ", logits.getShape())
 					print("Newlabels: ", new_labels_batch)
 					print("labels size: ", new_labels_batch.size())
+					print("labels shape: ", new_labels_batch.getShape())
 
 					input = torch.autograd.Variable(logits.view(-1, 2).to("cuda"))
 					target = torch.nn.functional.softmax(torch.autograd.Variable(new_labels_batch.view(-1).to("cuda")))