fix all tests but one

huggingface · Jun 29, 2020 · 8185265 · 8185265
1 parent c48bf39
commit 8185265
Showing 1 changed file with 0 additions and 3 deletions.
diff --git a/src/transformers/modeling_longformer.py b/src/transformers/modeling_longformer.py
@@ -1552,9 +1552,6 @@ def forward(
 
         pooled_output = self.dropout(pooled_output)
         logits = self.classifier(pooled_output)
-        print(f"logits: {logits.shape}")
-        print(f"pooled_output: {pooled_output.shape}")
-        print(f"num_choices: {num_choices}")
         reshaped_logits = logits.view(-1, num_choices)
 
         outputs = (reshaped_logits,) + outputs[2:]  # add hidden states and attention if they are here