ce

Code/train.py

@@ -104,9 +104,9 @@ def train(model, name, seed,gradient_accumulation_steps,mixup, threshold, lambda
 					logits=model.classifier(span_output.detach()) #target_value?
 						
 					#print("logits: ", logits)
-					# print("logits shape: ", list(logits.shape))
+					print("logits shape: ", list(logits.shape))
 					# print("Newlabels: ", new_labels_batch)
-					# print("labels shape: ", list(new_labels_batch.shape))
+					print("labels shape: ", list(new_labels_batch.shape))
 
 					logits = logits.view(-1, 2).to("cuda")
 					print("logits: ", logits)