mirror of
https://github.com/dragonpilot/dragonpilot.git
synced 2026-07-21 12:52:08 +08:00
Vehicle Researcher
dd778596b7
date: 2025-03-15T21:10:51 master commit: fb7b9c0f9420d228f03362970ebcfb7237095cf3
9 lines
263 B
Python
9 lines
263 B
Python
#!/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()
|