set requires_grad to None (#387)

* set requires_grad to None

* some things need gradients

* hmm, why was get_parameters filtering
This commit is contained in:
George Hotz
2022-09-21 11:16:02 -04:00
committed by GitHub
parent 29ae21bb0d
commit 271446e3eb
7 changed files with 26 additions and 17 deletions

View File

@@ -29,8 +29,8 @@ python3 setup.py develop
```python
from tinygrad.tensor import Tensor
x = Tensor.eye(3)
y = Tensor([[2.0,0,-2.0]])
x = Tensor.eye(3, requires_grad=True)
y = Tensor([[2.0,0,-2.0]], requires_grad=True)
z = y.matmul(x).sum()
z.backward()