Skip to content

Commit

Permalink
hotfix: cast mnist to float
Browse files Browse the repository at this point in the history
  • Loading branch information
geohot committed Apr 10, 2024
1 parent fea774f commit 216eb23
Showing 1 changed file with 3 additions and 0 deletions.
3 changes: 3 additions & 0 deletions examples/beautiful_mnist.py
Original file line number Diff line number Diff line change
Expand Up @@ -21,6 +21,9 @@ def __call__(self, x:Tensor) -> Tensor: return x.sequential(self.layers)
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))

Expand Down

0 comments on commit 216eb23

Please sign in to comment.