Skip to content

Commit

Permalink
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
clean up comments
Browse files Browse the repository at this point in the history
sichu2023 committed Dec 27, 2024
1 parent 4c77f97 commit fbc1b59
Showing 1 changed file with 0 additions and 2 deletions.
2 changes: 0 additions & 2 deletions sub-packages/bionemo-llm/src/bionemo/llm/lightning.py
Original file line number Diff line number Diff line change
@@ -294,10 +294,8 @@ def __init__(
self.log_train_ppl = log_train_ppl
self.log_val_ppl = log_val_ppl
if log_train_ppl:
# self.train_ppl = MegatronPerplexityMetric(ignore_index=-100)
self.train_ppl = torchmetrics.text.Perplexity(ignore_index=-100)
if log_val_ppl:
# self.valid_ppl = MegatronPerplexityMetric(ignore_index=-100)
self.valid_ppl = torchmetrics.text.Perplexity(ignore_index=-100)

def configure_model(self) -> None:

0 comments on commit fbc1b59

Please sign in to comment.