Skip to content

Commit 825c2db

Browse files
committed
Add message about looking into batch size related result changes
Signed-off-by: John St John <jstjohn@nvidia.com>
1 parent a2bdc85 commit 825c2db

1 file changed

Lines changed: 2 additions & 0 deletions

File tree

sub-packages/bionemo-evo2/tests/bionemo/evo2/run/test_predict.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -352,6 +352,8 @@ def test_predict_evo2_runs_with_log_probs(
352352
assert len(seq_idx_map) == num_sequences
353353
for original_idx, log_probs in zip(preds["seq_idx"], preds["log_probs_seqs"]):
354354
if mp_size > 1 or fp8:
355+
# FIXME changing batch size so it doesn't match also required dropping rel=1e-6 to rel=1e-3.
356+
# This should be investigated.
355357
rel = 1e-3
356358
else:
357359
rel = 1e-6

0 commit comments

Comments
 (0)