github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-10 23:48:01 -05:00
Code Issues Packages Projects Releases Wiki Activity

no tqdm in actions

Browse Source
This commit is contained in:
George Hotz
2020-10-25 16:40:08 -07:00
parent ad48061927
commit 2333c4dea7
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)):
for i in (t := trange(steps, disable=os.getenv('GITHUB_ACTIONS') 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))