mirror of
https://github.com/tinygrad/tinygrad.git
synced 2026-01-09 15:08:02 -05:00
fix handcode_opt bert [pr] (#8509)
* fix handcode_opt bert [pr] * too slow
This commit is contained in:
@@ -50,7 +50,7 @@ def get_sched_bert():
|
||||
# ignore grad norm and loss scaler for now
|
||||
loss.backward()
|
||||
targets += [x for x in optim.schedule_step()]
|
||||
sched = Tensor.schedule(targets)
|
||||
sched = Tensor.schedule(*targets)
|
||||
print(f"schedule length {len(sched)}")
|
||||
return sched
|
||||
|
||||
|
||||
Reference in New Issue
Block a user