Skip to content

Commit 5b986bd

Browse files
committed
update testing reference val
Signed-off-by: Sangkug Lym <slym@nvidia.com>
1 parent 220ff8a commit 5b986bd

1 file changed

Lines changed: 4 additions & 4 deletions

File tree

tests/collections/common/test_perf_metrics.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -93,10 +93,10 @@ def model_config(cfg):
9393
@pytest.mark.parametrize(
9494
"cfg, model_name, train_step_time, expected_value",
9595
[
96-
(LLAMA2_CFG_STR, None, 8, 377.53),
97-
(LLAMA2_CFG_STR, "llama2", 8, 377.53),
98-
(LLAMA2_CFG_STR, None, [8, 8, 8, 8], 377.53),
99-
(NEMOTRON_CFG_STR, None, 1.31, 642.73),
96+
(LLAMA2_CFG_STR, None, 8, 351.14),
97+
(LLAMA2_CFG_STR, "llama2", 8, 351.14),
98+
(LLAMA2_CFG_STR, None, [8, 8, 8, 8], 351.14),
99+
(NEMOTRON_CFG_STR, None, 1.31, 602.42),
100100
(
101101
UNSUPPORTED_MODEL_CFG_STR,
102102
None,

0 commit comments

Comments
 (0)