remove duplicate padding calculations in collate fn (#4617)

allenai · Sep 1, 2020 · 319794a · 319794a
1 parent de9165e
commit 319794a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/allennlp/data/data_loaders/data_loader.py b/allennlp/data/data_loaders/data_loader.py
@@ -20,7 +20,7 @@ def allennlp_collate(instances: List[Instance]) -> TensorDict:
     batch.
     """
     batch = Batch(instances)
-    return batch.as_tensor_dict(batch.get_padding_lengths())
+    return batch.as_tensor_dict()
 
 
 class DataLoader(Registrable):