update test

mlcommons · Oct 3, 2024 · 107c6b6 · 107c6b6
1 parent f15e227
commit 107c6b6
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/reference_algorithm_tests.py b/tests/reference_algorithm_tests.py
@@ -471,6 +471,7 @@ def _test_submission(workload_name,
  batch=batch,
  loss_type=workload.loss_type,
  optimizer_state=optimizer_state,
+ train_state={},
  eval_results=[],
  global_step=global_step,
  rng=update_rng)