From 67d9fc50d917c63cf67281106214e1d9ae018dff Mon Sep 17 00:00:00 2001 From: Sam Shleifer Date: Thu, 17 Sep 2020 18:32:31 -0400 Subject: [PATCH] [s2s] remove double assert (#7223) --- examples/seq2seq/finetune.py | 5 ----- 1 file changed, 5 deletions(-) diff --git a/examples/seq2seq/finetune.py b/examples/seq2seq/finetune.py index 7372139f99..4835e58439 100644 --- a/examples/seq2seq/finetune.py +++ b/examples/seq2seq/finetune.py @@ -102,11 +102,6 @@ class SummarizationModule(BaseTransformer): } assert self.target_lens["train"] <= self.target_lens["val"], f"target_lens: {self.target_lens}" assert self.target_lens["train"] <= self.target_lens["test"], f"target_lens: {self.target_lens}" - - if self.hparams.sortish_sampler and self.hparams.gpus > 1: - raise AssertionError("Sortish Sampler does not work for multigpu") - if self.hparams.sortish_sampler and self.hparams.max_tokens_per_batch is not None: - raise AssertionError("max tokens per batch and sortish sampler are incompatible.") if self.hparams.freeze_embeds: self.freeze_embeds() if self.hparams.freeze_encoder: