fix handcode_opt.py for bert (#6756)

2026-01-09 15:08:02 -05:00 · 2024-09-26 00:20:24 -04:00
parent 396c96357b
commit 0424c4967d
1 changed files with 1 additions and 1 deletions
--- a/examples/handcode_opt.py
+++ b/examples/handcode_opt.py
@@ -53,7 +53,7 @@ def get_sched_bert():
      # ignore grad norm and loss scaler for now
      loss.backward()
      targets += [x.lazydata for x in optim.schedule_step()]
-    sched = create_schedule(targets, seen)
+    sched = create_schedule(targets)
    print(f"schedule length {len(sched)}")
  return sched