Skip to content

GitLab

Unverified Commit 67d9fc50 authored by Sam Shleifer's avatar Sam Shleifer Committed by GitHub
Browse files

[s2s] remove double assert (#7223)

parent edbaad2c
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 5 deletions
examples/seq2seq/finetune.py
View file @ 67d9fc50
...@@ -102,11 +102,6 @@ class SummarizationModule(BaseTransformer): ...@@ -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["val"], f"target_lens: {self.target_lens}"
assert self.target_lens["train"] <= self.target_lens["test"], 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: if self.hparams.freeze_embeds:
self.freeze_embeds() self.freeze_embeds()
if self.hparams.freeze_encoder: if self.hparams.freeze_encoder:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment