We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 5ecef8b commit 3f4705cCopy full SHA for 3f4705c
.github/workflows/baselines/test_maxtext_metrics.py
@@ -31,6 +31,7 @@ def test_loss(baseline_filename):
31
loss_expected = {step: loss_expected_values[i] for i, step in enumerate(
32
range(start_step, end_step+1, interval))}
33
loss_actual = test_utils.read_maxtext_tb_tag(event_file, loss_summary_name)
34
+ del loss_actual[0] # removing the very first step
35
assert loss_expected.keys() == loss_actual.keys(), \
36
f"Steps at which loss was emitted for run do not match baseline. \
37
Actual steps: {loss_actual.keys()}, Baseline steps: {loss_expected.keys()}"
0 commit comments