remove pp error

Signed-off-by: sichu <[email protected]>
NVIDIA · Jan 17, 2025 · a404cc5 · a404cc5
1 parent e7191c8
commit a404cc5
Showing 1 changed file with 0 additions and 3 deletions.
diff --git a/sub-packages/bionemo-esm2/src/bionemo/esm2/scripts/train_esm2.py b/sub-packages/bionemo-esm2/src/bionemo/esm2/scripts/train_esm2.py
@@ -283,9 +283,6 @@ def main(
     if scheduler_num_steps is None:
         scheduler_num_steps = num_steps
 
-    if (log_train_ppl or log_val_ppl) and pipeline_model_parallel_size > 1:
-        raise NotImplementedError("Perplexity logging does not support pipeline parallelism yet.")
-
     model = biobert_lightning_module(
         esm2_config,
         tokenizer=tokenizer,