[wip/s2s] DistributedSortishSampler (#7056)

2020-09-10 15:23:44 -04:00
parent 514486739c
commit 77950c485a
3 changed files with 79 additions and 24 deletions
--- a/examples/seq2seq/test_seq2seq_examples.py
+++ b/examples/seq2seq/test_seq2seq_examples.py
@@ -149,9 +149,9 @@ class TestSummarizationDistiller(unittest.TestCase):
            no_teacher=True,
            freeze_encoder=True,
            gpus=2,
-            sortish_sampler=False,
+            sortish_sampler=True,
        )
-        self._test_distiller_cli(updates)
+        self._test_distiller_cli(updates, check_contents=False)

    def test_distill_no_teacher(self):
        updates = dict(student_encoder_layers=2, student_decoder_layers=1, no_teacher=True)