mirrors/tinygrad
2
0
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-06-24 02:14:17 +00:00
Code Issues Projects Releases Packages Wiki Activity

seems more generic

Browse source
This commit is contained in:
George Hotz 2020-10-25 16:40:37 -07:00
commit ea41f5e1c1
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
test/test_mnist.py
View file

@ -39,7 +39,7 @@ class TinyConvNet:
def train(model, optim, steps, BS=128):
losses, accuracies = [], []
for i in (t := trange(steps, disable=os.getenv('GITHUB_ACTIONS') is not None)):
for i in (t := trange(steps, disable=os.getenv('CI') is not None)):
samp = np.random.randint(0, X_train.shape[0], size=(BS))
x = Tensor(X_train[samp].reshape((-1, 28*28)).astype(np.float32))