github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-09 15:08:02 -05:00
Code Issues Packages Projects Releases Wiki Activity

remove Tensor.no_grad, it's meaningless now [pr] (#10556)

Browse Source
This commit is contained in:
George Hotz
2025-05-28 22:20:02 -07:00
committed by GitHub
parent e4e7b5d7e1
commit b3b43a82c4
35 changed files with 17 additions and 80 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
extra/optimization/rl.py
View File

@@ -18,7 +18,7 @@ if __name__ == "__main__":
# select a world
all_feats, all_acts, all_rews = [], [], []
while 1:
Tensor.no_grad, Tensor.training = True, False
Tensor.training = False
lin = ast_str_to_lin(random.choice(ast_strs))
rawbufs = bufs_from_lin(lin)
tm = last_tm = base_tm = time_linearizer(lin, rawbufs)
@@ -63,7 +63,7 @@ if __name__ == "__main__":
BS = 32
if len(all_feats) >= BS:
Tensor.no_grad, Tensor.training = False, True
Tensor.training = True
x = Tensor(all_feats[:BS])
mask = np.zeros((BS, len(actions)+1), dtype=np.float32)
mask[range(BS), all_acts[:BS]] = all_rews[:BS]