You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
9 lines
263 B
9 lines
263 B
6 days ago
|
#!/bin/bash
|
||
|
|
from tinygrad import Tensor
|
||
|
|
from extra.datasets import fetch_mnist
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
X_train, Y_train, X_test, Y_test = fetch_mnist(tensors=True)
|
||
|
|
samples = Tensor.randint(512, high=X_train.shape[0])
|
||
|
|
select = X_train[samples].realize()
|