mirrors/tinygrad
2
0
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-06-24 02:14:17 +00:00
Code Issues Projects Releases Packages Wiki Activity

hotfix: cast mnist to float

Browse source
This commit is contained in:
George Hotz 2024-04-09 19:30:03 -07:00
commit 216eb235e5
1 changed files with 3 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

3
examples/beautiful_mnist.py
View file

@ -21,6 +21,9 @@ class Model:
if __name__ == "__main__":
X_train, Y_train, X_test, Y_test = mnist()
# TODO: remove this when HIP is fixed
X_train, X_test = X_train.float(), X_test.float()
model = Model()
opt = nn.optim.Adam(nn.state.get_parameters(model))