mawei/tinygrad
1
0
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-06-13 08:28:55 +08:00
Code Issues Packages Projects Releases Wiki Activity

seems more generic

Browse Source
This commit is contained in:
George Hotz
2020-10-25 16:40:37 -07:00
parent 2333c4dea7
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))