From 83530a585fc9bdabc4a6a4e1280e2e5da1a7eacd Mon Sep 17 00:00:00 2001 From: George Hotz Date: Sat, 2 Mar 2024 05:38:32 -0800 Subject: [PATCH] add quick external data select test --- test/external/external_test_mnist_data_select.py | 8 ++++++++ 1 file changed, 8 insertions(+) create mode 100644 test/external/external_test_mnist_data_select.py diff --git a/test/external/external_test_mnist_data_select.py b/test/external/external_test_mnist_data_select.py new file mode 100644 index 0000000000..d4fd6c0a57 --- /dev/null +++ b/test/external/external_test_mnist_data_select.py @@ -0,0 +1,8 @@ +#!/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()