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

hotfix convert Y_train to int before passing into index (#2850)

Browse Source
This commit is contained in:
chenyu
2023-12-19 11:40:56 -05:00
committed by GitHub
parent fec8e9060c
commit 6d7e9e0a56
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/hlb_cifar10.py
View File

@@ -263,7 +263,7 @@ def train_cifar():
X_train, X_test = X_train.to(device=Device.DEFAULT).float(), X_test.to(device=Device.DEFAULT).float()
Y_train, Y_test = Y_train.to(device=Device.DEFAULT).float(), Y_test.to(device=Device.DEFAULT).float()
# one-hot encode labels
Y_train, Y_test = Tensor.eye(10)[Y_train], Tensor.eye(10)[Y_test]
Y_train, Y_test = Tensor.eye(10)[Y_train.cast(dtypes.int32)], Tensor.eye(10)[Y_test.cast(dtypes.int32)]
# preprocess data
X_train, X_test = X_train.sequential(transform), X_test.sequential(transform)