We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 220ff8a commit 5b986bdCopy full SHA for 5b986bd
1 file changed
tests/collections/common/test_perf_metrics.py
@@ -93,10 +93,10 @@ def model_config(cfg):
93
@pytest.mark.parametrize(
94
"cfg, model_name, train_step_time, expected_value",
95
[
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),
+ (LLAMA2_CFG_STR, None, 8, 351.14),
+ (LLAMA2_CFG_STR, "llama2", 8, 351.14),
+ (LLAMA2_CFG_STR, None, [8, 8, 8, 8], 351.14),
+ (NEMOTRON_CFG_STR, None, 1.31, 602.42),
100
(
101
UNSUPPORTED_MODEL_CFG_STR,
102
None,
0 commit comments