remove Tensor.test() in retinanet (#10770)

test was removed
2026-01-09 15:08:02 -05:00 · 2025-06-10 19:14:57 -07:00
parent 25304c3dd0
commit 81e296d7b8
1 changed files with 1 additions and 1 deletions
--- a/examples/mlperf/model_train.py
+++ b/examples/mlperf/model_train.py
@@ -608,7 +608,7 @@ def train_retinanet():

      if getenv("RESET_STEP", 1): _train_step.reset()

-      with Tensor.train(mode=False), Tensor.test():
+      with Tensor.train(mode=False):
        if not RUNMLPERF:
          i, proc = 0, _fake_data_get(EVAL_BS, val=(val:=True))
        else: